fix:check what else we edited

Author: stevenhua0320

src/diffpy/srmise/dataclusters.py

@@ -206,19 +206,19 @@ def __next__(self):
 
         if np.abs(nearest_cluster[1]) <= self.res:
             # Add to an existing cluster
-            self.lastcluster_idx = int(nearest_cluster[0])
-            if test_idx < self.clusters[int(nearest_cluster[0]), 0]:
-                self.clusters[int(nearest_cluster[0]), 0] = test_idx
+            self.lastcluster_idx = nearest_cluster[0]
+            if test_idx < self.clusters[nearest_cluster[0], 0]:
+                self.clusters[nearest_cluster[0], 0] = test_idx
             else:
-                self.clusters[int(nearest_cluster[0]), 1] = test_idx
+                self.clusters[nearest_cluster[0], 1] = test_idx
         else:
             # Make a new cluster
             if nearest_cluster[1] < 0:
                 # Insert left of nearest cluster
-                self.lastcluster_idx = int(nearest_cluster[0])
+                self.lastcluster_idx = nearest_cluster[0]
             else:
                 # insert right of nearest cluster
-                self.lastcluster_idx = int(nearest_cluster[0]) + 1
+                self.lastcluster_idx = nearest_cluster[0] + 1
             self.clusters = np.insert(
                 self.clusters,
                 int(self.lastcluster_idx),
@@ -289,7 +289,7 @@ def find_nearest_cluster(self, idx):
             return None
 
         flat_idx = clusters_flat.searchsorted(idx)
-        near_idx = flat_idx / 2
+        near_idx = flat_idx // 2
 
         if flat_idx == len(clusters_flat):
             # test_idx is right of the last cluster

src/diffpy/srmise/peakextraction.py

@@ -905,7 +905,7 @@ def extract_single(self, recursion_depth=1):
                 )
             else:
                 # Update an existing cluster
-                mclusters[int(step.lastcluster_idx)].change_slice(step.cut(step.lastcluster_idx))
+                mclusters[step.lastcluster_idx].change_slice(step.cut(step.lastcluster_idx))
 
             # Find newly adjacent clusters
             adjacent = step.find_adjacent_clusters().ravel()