oops, direct ptycho viz bug

py4dstem · Nov 15, 2024 · af0109a · af0109a
1 parent 7216112
commit af0109a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4DSTEM/process/phase/direct_ptychography.py b/py4DSTEM/process/phase/direct_ptychography.py
@@ -649,7 +649,7 @@ def preprocess(
                 self._reciprocal_sampling[0] * alpha_probe / self.angular_sampling[0]
             )
 
-            bf_inds = f[self._trotter_inds[0], self._trotter_inds[1]] < q_probe
+            bf_inds = f[self._trotter_inds[0], self._trotter_inds[1]] < q_probe**2
             low_ind_x = self._trotter_inds[0][bf_inds][0]
             low_ind_y = self._trotter_inds[1][bf_inds][0]
             high_ind_x = self._trotter_inds[0][~bf_inds][0]