Merge pull request py4dstem#565 from alex-rakowski/mask_hotpatch

fixing is to ==
bsavitzky · Nov 14, 2023 · 76ae0fa · 76ae0fa
2 parents e03f0a8 + da327ef
commit 76ae0fa
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 is True,
+                data_mask=mask == True,  # noqa E712
             )
             popt_y, pcov_y, qy0_fit, _ = fit_2D(
                 f,
                 qy0_meas,
                 robust=robust,
                 robust_steps=robust_steps,
                 robust_thresh=robust_thresh,
-                data_mask=mask is True,
+                data_mask=mask == True,  # noqa E712
             )
 
     # Compute residuals