bug in ptycho-tomo object support mask

py4dstem · Nov 16, 2024 · b195d0d · b195d0d
1 parent af0109a
commit b195d0d
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/py4DSTEM/process/phase/ptychographic_tomography.py b/py4DSTEM/process/phase/ptychographic_tomography.py
@@ -1233,7 +1233,10 @@ def reconstruct(
                     use_fourier_rotation,
                 )
 
-                if not collective_measurement_updates:
+                if (
+                    not collective_measurement_updates
+                    and object_real_space_support_mask is not None
+                ):
                     object_real_space_support_mask = self._rotate_zxy_volume(
                         object_real_space_support_mask,
                         rot_matrix @ old_rot_matrix.T,
@@ -1464,11 +1467,12 @@ def reconstruct(
                     use_fourier_rotation,
                 )
 
-                object_real_space_support_mask = self._rotate_zxy_volume(
-                    object_real_space_support_mask,
-                    old_rot_matrix,
-                    use_fourier_rotation,
-                )
+                if object_real_space_support_mask is not None:
+                    object_real_space_support_mask = self._rotate_zxy_volume(
+                        object_real_space_support_mask,
+                        old_rot_matrix,
+                        use_fourier_rotation,
+                    )
 
                 # object only
                 self._object = self._object_constraints(