[utils] Fix SimpleITK read_image(); force pull in read_polydata()

BioMedIA · Sep 4, 2023 · 4aad3a8 · 4aad3a8
1 parent 0ba0e88
commit 4aad3a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/deepali/utils/simpleitk/imageio.py b/src/deepali/utils/simpleitk/imageio.py
@@ -7,7 +7,7 @@
 def read_image(path: PathUri) -> sitk.Image:
     r"""Read image from file."""
     with StorageObject.from_path(path) as obj:
-        if not path.exists():
+        if not obj.is_file():
             raise FileNotFoundError(f"No such file or no access: '{path}'")
         obj.pull(force=True)
         image = sitk.ReadImage(str(obj.path))

diff --git a/src/deepali/utils/vtk/polydataio.py b/src/deepali/utils/vtk/polydataio.py
@@ -22,7 +22,7 @@ def read_polydata(path: PathUri) -> vtkPolyData:
     with StorageObject.from_path(path) as obj:
         if not obj.is_file():
             raise FileNotFoundError(f"File not found: {obj.uri}")
-        obj = obj.pull()
+        obj = obj.pull(force=True)
         if not obj.path.is_file():
             raise FileNotFoundError(f"File not found: {obj.path}")
         suffix = obj.path.suffix.lower()