fix: more fix

mind-inria · Feb 11, 2025 · 545ca47 · 545ca47
1 parent 0933df2
commit 545ca47
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/src/snake/core/handlers/fov.py b/src/snake/core/handlers/fov.py
@@ -143,7 +143,6 @@ def get_static(self, phantom: Phantom, sim_conf: SimConfig) -> Phantom:
             zoom_factor=zoom_factor,
         )
         if phantom.smaps is not None:
-            return Phantom.from_masks(new_masks, sim_conf)
             new_smaps = np.zeros(
                 (
                     phantom.smaps.shape[0],

diff --git a/src/snake/core/phantom/static.py b/src/snake/core/phantom/static.py
@@ -227,18 +227,19 @@ def to_mrd_dataset(
             ),
         )
         # Add the smaps
-        dataset.append_image(
-            "smaps",
-            mrd.image.Image(
-                head=mrd.image.ImageHeader(
-                    matrixSize=mrd.xsd.matrixSizeType(*self.smaps.shape[1:]),
-                    fieldOfView_mm=mrd.xsd.fieldOfViewMm(*sim_conf.fov_mm),
-                    channels=len(self.smaps),
-                    acquisition_time_stamp=0,
+        if self.smaps is not None:
+            dataset.append_image(
+                "smaps",
+                mrd.image.Image(
+                    head=mrd.image.ImageHeader(
+                        matrixSize=mrd.xsd.matrixSizeType(*self.smaps.shape[1:]),
+                        fieldOfView_mm=mrd.xsd.fieldOfViewMm(*sim_conf.fov_mm),
+                        channels=len(self.smaps),
+                        acquisition_time_stamp=0,
+                    ),
+                    data=self.smaps,
                 ),
-                data=self.smaps,
-            ),
-        )
+            )
         return dataset
 
     @classmethod