Merge pull request #28 from ArcanaFramework/develop

Touched up ITK union types
ArcanaFramework · Apr 24, 2024 · c707e59 · c707e59
2 parents 2faef26 + 8d0eae7
commit c707e59
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/fileformats/medimage/__init__.py b/fileformats/medimage/__init__.py
@@ -102,5 +102,6 @@
     MetaImage,
     Nrrd,
     NrrdGz,
-    ItkSupported,
+    ItkImage,
+    ItkAll,
 )
diff --git a/fileformats/medimage/itk.py b/fileformats/medimage/itk.py
@@ -44,6 +44,10 @@ class NrrdGz(Gzip[Nrrd]):
     alternate_exts = (".nhdr.gz",)
 
 
-ItkSupported = ty.Union[
+ItkImage = ty.Union[
+    Nifti1, NiftiGz, Dicom, Bitmap, Tiff, Jpeg, GIPL, MetaImage, Nrrd, NrrdGz, PGM
+]
+
+ItkAll = ty.Union[
     Nifti1, NiftiGz, Dicom, Bitmap, Tiff, Jpeg, GIPL, MetaImage, Nrrd, NrrdGz, PGM, VTK
 ]