fine, this should work

py4dstem · Oct 8, 2024 · 6c1d759 · 6c1d759
1 parent 9739a1e
commit 6c1d759
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
@@ -541,7 +541,7 @@ def preprocess(
                 / self.angular_sampling[0]
             )
 
-            bf_inds = f[*trotter_inds] < q_probe
+            bf_inds = f[trotter_inds[0], trotter_inds[1]] < q_probe
             low_ind_x = trotter_inds[0][bf_inds][0]
             low_ind_y = trotter_inds[1][bf_inds][0]
             high_ind_x = trotter_inds[0][~bf_inds][0]