fix(nii2dcm): fixed check for the number of dicom (#21)

deepc-health · Apr 13, 2022 · c3ec65d · c3ec65d
1 parent cad710f
commit c3ec65d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nekton/nii2dcm.py b/nekton/nii2dcm.py
@@ -48,8 +48,8 @@ def _check_all_dicoms(self, dcmfiles: List[Path], seg: np.ndarray) -> List[Path]
         Returns:
             List[Path]: sorted list of dicoms based on the order
         """
-        assert seg.shape[-1] == len(
-            dcmfiles
+        assert len(dcmfiles) in list(
+            seg.shape
         ), f"""Need 1 DICOM per slice of NifTi;
         Found {len(dcmfiles)} DICOMS for {len(seg)} NifTi slice"""