C409 C405

alex-rakowski · Nov 10, 2023 · 4288ca2 · 4288ca2
1 parent 7892191
commit 4288ca2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/py4DSTEM/process/calibration/ellipse.py b/py4DSTEM/process/calibration/ellipse.py
@@ -199,7 +199,7 @@ def fit_ellipse_amorphous_ring(data, center, fitradii, p0=None, mask=None):
     A, B, C = convert_ellipse_params_r(R, R, 0)
 
     # Populate initial parameters
-    p0_guess = tuple([I0, I1, sigma0, sigma1, sigma2, c_bkgd, x0, y0, A, B, C])
+    p0_guess = (I0, I1, sigma0, sigma1, sigma2, c_bkgd, x0, y0, A, B, C)
     if p0 is None:
         _p0 = p0_guess
     else:

diff --git a/py4DSTEM/process/classification/braggvectorclassification.py b/py4DSTEM/process/classification/braggvectorclassification.py
@@ -856,7 +856,7 @@ def get_initial_classes(
         seed_new_class = True
         while seed_new_class:
             ind1, ind2 = np.unravel_index(np.argmax(n_point_function), (N, N))
-            BP_set = set([ind1, ind2])
+            BP_set = {ind1, ind2}
             grow_class = True
             while grow_class:
                 frequencies = np.zeros(N)
@@ -910,7 +910,7 @@ def get_initial_classes(
         seed_new_class = True
         while seed_new_class:
             ind1, ind2, ind3 = np.unravel_index(np.argmax(n_point_function), (N, N, N))
-            BP_set = set([ind1, ind2, ind3])
+            BP_set = {ind1, ind2, ind3}
             grow_class = True
             while grow_class:
                 frequencies = np.zeros(N)