Merge pull request #30 from khanlab/uni-images

fix for UNI_Images
khanlab · May 2, 2022 · 9613122 · 9613122
2 parents 9ca437e + 7a32e8d
commit 9613122
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/heuristics/cfmm_base.py b/heuristics/cfmm_base.py
@@ -339,11 +339,11 @@ def infotodict(seqinfo):
                 elif ('DIS3D' in (s.image_type[4].strip())):
                     info[DIS3D_t1w].append({'item': s.series_id})
             if ('UNI_Images' in (s.series_description).strip()):
-                if ('DIS2D' in (s.image_type[3].strip())):
+                if ('DIS2D' in s.image_type):
                     info[DIS2D_uni_mp2rage].append({'item': s.series_id})
-                if ('DIS3D' in (s.image_type[3].strip())):
+                if ('DIS3D' in s.image_type):
                     info[DIS3D_uni_mp2rage].append({'item': s.series_id})
-                if ('ND' in (s.image_type[3].strip())):
+                if ('ND' in s.image_type):
                     info[uni_mp2rage].append({'item': s.series_id})
             if ('_INV2' in (s.series_description).strip()):
                 if ('DIS2D' in (s.image_type[3].strip())):