move progress update step

py4dstem · Oct 5, 2023 · a48093f · a48093f
1 parent 221b34b
commit a48093f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/py4DSTEM/process/diffraction/crystal_ACOM.py b/py4DSTEM/process/diffraction/crystal_ACOM.py
@@ -1826,10 +1826,6 @@ def cluster_grains(
             search = False
 
         else:
-            new_marks = mark_total - np.sum(np.max(mark, axis=2))
-            pbar.update(new_marks)
-            mark_total -= new_marks
-
             # Start cluster
             x, y, z = np.unravel_index(inds_grain, sig.shape)
             mark[x, y, z] = False
@@ -1909,6 +1905,11 @@ def cluster_grains(
             self.cluster_orientation.append(orientation_cluster)
             self.cluster_inds.append(xyg)
 
+            # update progressbar
+            new_marks = mark_total - np.sum(np.max(mark, axis=2))
+            pbar.update(new_marks)
+            mark_total -= new_marks
+
     pbar.close()