Fix tqdm postfix

flatironinstitute · Feb 18, 2024 · 3a39265 · 3a39265
1 parent 5f806c7
commit 3a39265
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/inferelator_velocity/utils/noise2self.py b/inferelator_velocity/utils/noise2self.py
@@ -185,6 +185,7 @@ def knn_noise2self(
 
         # Update the progress bar
         tqdm_pbar.postfix = f"{pc} PCs Neighbor Search"
+        tqdm_pbar.update(0)
 
         # Search through the neighbors space
         mses[i, :] = _search_k(
@@ -343,6 +344,7 @@ def _search_k(
 
         if hasattr(pbar, 'postfix'):
             pbar.postfix = _postfix + f" ({k[i]} N)"
+            pbar.update(0)
 
         # Extract k non-zero neighbors from the graph
         k_graph = local_optimal_knn(