Readability updates for preprocessing.

mist-medical · Sep 26, 2024 · 830af4e · 830af4e
1 parent 13f223b
commit 830af4e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mist/preprocess_data/preprocess.py b/mist/preprocess_data/preprocess.py
@@ -394,7 +394,9 @@ def preprocess_example(
 
         # Put mask into standard space
         mask = ants.reorient_image2(mask, "RAI")
-        mask.set_direction(np.eye(3))
+        mask.set_direction(
+            preprocessing_constants.PreprocessingConstants.RAI_ANTS_DIRECTION
+        )
         mask = resample_mask(
             mask,
             labels=config["labels"],