remove equal datacube shapes check for magnetic ptychography

py4dstem · Aug 10, 2024 · 39ba85e · 39ba85e
1 parent e8c04ac
commit 39ba85e
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/py4DSTEM/process/phase/magnetic_ptychography.py b/py4DSTEM/process/phase/magnetic_ptychography.py
@@ -377,10 +377,6 @@ def preprocess(
                 f"datacube must be the same length as magnetic_contribution_sign, not length {len(self._datacube)}."
             )
 
-        dc_shapes = [dc.shape for dc in self._datacube]
-        if dc_shapes.count(dc_shapes[0]) != self._num_measurements:
-            raise ValueError("datacube intensities must be the same size.")
-
         if self._positions_mask is not None:
             self._positions_mask = np.asarray(self._positions_mask, dtype="bool")
 

diff --git a/py4DSTEM/process/phase/xray_magnetic_ptychography.py b/py4DSTEM/process/phase/xray_magnetic_ptychography.py
@@ -375,10 +375,6 @@ def preprocess(
                 f"datacube must be the same length as magnetic_contribution_sign, not length {len(self._datacube)}."
             )
 
-        dc_shapes = [dc.shape for dc in self._datacube]
-        if dc_shapes.count(dc_shapes[0]) != self._num_measurements:
-            raise ValueError("datacube intensities must be the same size.")
-
         if self._positions_mask is not None:
             self._positions_mask = np.asarray(self._positions_mask, dtype="bool")