Revert "Merge pull request #565 from alex-rakowski/mask_hotpatch"

This reverts commit 76ae0fa, reversing changes made to e03f0a8.
py4dstem · Nov 16, 2023 · 8a26648 · 8a26648
1 parent ac9c0d1
commit 8a26648
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py4DSTEM/process/calibration/origin.py b/py4DSTEM/process/calibration/origin.py
@@ -154,15 +154,15 @@ def fit_origin(
                 robust=robust,
                 robust_steps=robust_steps,
                 robust_thresh=robust_thresh,
-                data_mask=mask == True,  # noqa E712
+                data_mask=mask is True,
             )
             popt_y, pcov_y, qy0_fit, _ = fit_2D(
                 f,
                 qy0_meas,
                 robust=robust,
                 robust_steps=robust_steps,
                 robust_thresh=robust_thresh,
-                data_mask=mask == True,  # noqa E712
+                data_mask=mask is True,
             )
 
     # Compute residuals