fixing is to ==

py4dstem · Nov 14, 2023 · da327ef · da327ef
1 parent e03f0a8
commit da327ef
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