Add t1c to the list of possible modalities (#613)

mlcommons · Sep 4, 2024 · f6927b8 · f6927b8
1 parent 7348aca
commit f6927b8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/examples/BraTS/data_prep/mlcube/mlcube.yaml b/examples/BraTS/data_prep/mlcube/mlcube.yaml
@@ -8,7 +8,7 @@ platform:
 
 docker:
   # Image name.
-  image: mlcommons/fets_data-prep
+  image: mlcommons/fets_data-prep-v2
   # Docker build context relative to $MLCUBE_ROOT. Default is `build`.
   build_context: "../project"
   # Docker file name within docker build context, default is `Dockerfile`.

diff --git a/examples/BraTS/data_prep/project/sanity_check.py b/examples/BraTS/data_prep/project/sanity_check.py
@@ -17,6 +17,7 @@ def check_subject_validity(subject_dir):
     subject_valid = True
     strings_to_check = [
         "_t1.nii.gz",
+        "_t1c.nii.gz",
         "_t1ce.nii.gz",
         "_t2.nii.gz",
         "_flair.nii.gz",