Move logging statement

open2c · Mar 20, 2024 · 9c9f864 · 9c9f864
1 parent 68a90b0
commit 9c9f864
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/cooler/reduce.py b/src/cooler/reduce.py
@@ -183,9 +183,8 @@ def __iter__(self) -> Iterator[dict[str, np.ndarray]]:
         logger.info(f"nnzs: {nnzs}")
 
         starts = [0] * len(self.coolers)
-        for bp in bin1_partition[1:]:
-            stops = [index[bp] for index in indexes]
-            logger.info(f"records consumed: {stops}")
+        for bin1_id in bin1_partition[1:]:
+            stops = [index[bin1_id] for index in indexes]
 
             # extract, concat
             combined = pd.concat(
@@ -207,6 +206,7 @@ def __iter__(self) -> Iterator[dict[str, np.ndarray]]:
 
             yield {k: v.values for k, v in df.items()}
 
+            logger.info(f"records consumed: {stops}")
             starts = stops