relax nibabel requirement

kaiko-ai · Nov 26, 2024 · b1208ad · b1208ad
1 parent 4129e34
commit b1208ad
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 12 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -59,7 +59,7 @@ file = "LICENSE"
 [project.optional-dependencies]
 vision = [
     "h5py>=3.10.0",
-    "nibabel>=5.2.0",
+    "nibabel>=4.0.1",
     "opencv-python-headless>=4.9.0.80",
     "timm>=1.0.9",
     "torchvision>=0.17.0",
@@ -72,7 +72,7 @@ vision = [
 ]
 all = [
     "h5py>=3.10.0",
-    "nibabel>=5.2.0",
+    "nibabel>=4.0.1",
     "opencv-python-headless>=4.9.0.80",
     "timm>=1.0.9",
     "torchvision>=0.17.0",

diff --git a/src/eva/vision/utils/io/nifti.py b/src/eva/vision/utils/io/nifti.py
@@ -54,7 +54,6 @@ def save_array_as_nifti(
         dtype: The data type to save the image.
     """
     nifti_image = nib.Nifti1Image(array, affine=np.eye(4), dtype=dtype)  # type: ignore
-    nifti_image.header.get_xyzt_units()
     nifti_image.to_filename(filename)