simplified progress bar in terahac loop

janreising · May 30, 2024 · 9875907 · 9875907
1 parent 1936dce
commit 9875907
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/astrocast/clustering.py b/astrocast/clustering.py
@@ -2680,7 +2680,7 @@ def run_terahac(self, similarity_matrix: np.ndarray,
 
                 # Collect merges in sub graphs
                 merges = []
-                if not parallel:
+                if not parallel or len(sub_graphs) < 2:
                     for subgraph in sub_graphs:
                         merges.extend(self.subgraph_hac(subgraph))
                 else:
@@ -2733,7 +2733,7 @@ def run_terahac(self, similarity_matrix: np.ndarray,
                     break
 
                 counter += 1
-                pbar.update(-len(to_prune) - len(merges))
+                pbar.update(len(to_prune) + len(merges))
 
             if counter >= stop_after:
                 tqdm.write(f"Prematurely stopped run after {counter} iterations.")