numpy indexing changed?

py4dstem · Oct 8, 2024 · 20d8359 · 20d8359
1 parent 52d29a8
commit 20d8359
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/py4DSTEM/process/phase/direct_ptychography.py b/py4DSTEM/process/phase/direct_ptychography.py
@@ -540,10 +540,10 @@ def preprocess(
                 * self._semiangle_cutoff
                 / self.angular_sampling[0]
             )
-            low_ind_x = trotter_inds[0][(f[*trotter_inds] < q_probe)][0]
-            low_ind_y = trotter_inds[1][(f[*trotter_inds] < q_probe)][0]
-            high_ind_x = trotter_inds[0][(f[*trotter_inds] > q_probe)][0]
-            high_ind_y = trotter_inds[1][(f[*trotter_inds] > q_probe)][0]
+            low_ind_x = trotter_inds[0][f[*trotter_inds] < q_probe][0]
+            low_ind_y = trotter_inds[1][f[*trotter_inds] < q_probe][0]
+            high_ind_x = trotter_inds[0][f[*trotter_inds] > q_probe][0]
+            high_ind_y = trotter_inds[1][f[*trotter_inds] > q_probe][0]
 
             figsize = kwargs.pop("figsize", (13, 4))
             chroma_boost = kwargs.pop("chroma_boost", 1)