fix(tests): missing os

jjjermiah · Jan 30, 2024 · 90e8e2f · 90e8e2f
1 parent 2d5bf9e
commit 90e8e2f
Showing 1 changed file with 21 additions and 26 deletions.
diff --git a/src/nbiatoolkit/nbia.py b/src/nbiatoolkit/nbia.py
@@ -1,4 +1,3 @@
-from math import e
 from .auth import OAuth2
 from .utils.nbia_endpoints import NBIA_ENDPOINTS
 from .logger.logger import setup_logger
@@ -147,13 +146,10 @@ def getPatients(self, Collection: str = "") -> Union[list[dict[str, str]], None]
             ), "PatientId must be a string, but received: %s" % type(
                 patient["PatientId"]
             )
+
+
             patientList.append(
-                {
-                    "PatientId": patient["PatientId"],
-                    "PatientName": patient["PatientName"],
-                    "PatientSex": patient["PatientSex"],
-                    "Collection": patient["Collection"],
-                }
+                patient
             )
 
         return patientList
@@ -269,27 +265,26 @@ def downloadSeries(
 
             try:
                 os.makedirs(downloadDir)
+            except FileExistsError:
+                pass
+
+            for seriesUID in SeriesInstanceUID:
+                future = executor.submit(
+                    self._downloadSingleSeries,
+                    SeriesInstanceUID=seriesUID,
+                    downloadDir=downloadDir,
+                    filePattern=filePattern,
+                    overwrite=overwrite,
+                )
+                futures.append(future)
 
-                for seriesUID in SeriesInstanceUID:
-                    future = executor.submit(
-                        self._downloadSingleSeries,
-                        SeriesInstanceUID=seriesUID,
-                        downloadDir=downloadDir,
-                        filePattern=filePattern,
-                        overwrite=overwrite,
-                    )
-                    futures.append(future)
+            # Use tqdm to create a progress bar
+            with tqdm(
+                total=len(futures), desc=f"Downloading {len(futures)} series"
+            ) as pbar:
+                for future in cf.as_completed(futures):
+                    pbar.update(1)
 
-            except Exception as e:
-                self.log.error("Error creating download directory: %s", e)
-                raise e
-            else:
-                # Use tqdm to create a progress bar
-                with tqdm(
-                    total=len(futures), desc=f"Downloading {len(futures)} series"
-                ) as pbar:
-                    for future in cf.as_completed(futures):
-                        pbar.update(1)
             return True
 
     # _downloadSingleSeries is a helper function that downloads a single series