fix typing errors

jjjermiah · Jan 29, 2024 · ab50dac · ab50dac
1 parent 1913844
commit ab50dac
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/nbiatoolkit/dicomsort/dicomsort.py b/src/nbiatoolkit/dicomsort/dicomsort.py
@@ -83,11 +83,12 @@ def sortSingleDICOMFile(
 
         os.makedirs(os.path.dirname(targetFilename), exist_ok=True)
 
-        match option:
-            case "copy":
-                shutil.copyfile(src=filePath, dst=targetFilename)
-            case "move":
-                shutil.move(src=filePath, dst=targetFilename)
+        if option == "copy":
+            shutil.copyfile(src=filePath, dst=targetFilename)
+        elif option == "move":
+            shutil.move(src=filePath, dst=targetFilename)
+        else:
+            raise ValueError(f"Invalid option: {option}")
 
         return True
 
@@ -101,7 +102,7 @@ def sortDICOMFiles(self, option: str = "copy", overwrite: bool = False) -> bool:
 
         return all(results)
 
-    def _get_dicom_files(self) -> list[str]:
+    def _get_dicom_files(self) -> 'list[str]':
         dicom_file_paths = []
         # Iterate over all files in the source directory
         for root, dirs, files in os.walk(self.sourceDir):