Merge pull request #8 from brainsciencecenter/master

naterichman · web-flow · commit f5a7ab7ea3ae · 2021-11-16T10:09:05.000-06:00
ENH: Add dwi regexes to match HCP and ABCD
diff --git a/classification_from_label.py b/classification_from_label.py
@@ -114,6 +114,8 @@ def is_anatomy(label):
 # Diffusion
 def is_diffusion(label):
     regexes = [
+        re.compile('^dmri_', re.IGNORECASE),
+        re.compile('_dmri$', re.IGNORECASE),
         re.compile('dti', re.IGNORECASE),
         re.compile('dwi', re.IGNORECASE),
         re.compile('diff_', re.IGNORECASE),
