Updated generate_boundaries in prediciton.boundary

EliHei2 · Oct 24, 2024 · b1da3e6 · b1da3e6
1 parent d8083f9
commit b1da3e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/segger/prediction/boundary.py b/src/segger/prediction/boundary.py
@@ -328,7 +328,7 @@ def get_cycles(graph: dict):
 def generate_boundaries(df, x="x_location", y="y_location", cell_id="segger_cell_id"):
     res = []
     group_df = df.groupby(cell_id)
-    for cell_id, t in tqdm(group_df, total=len(group_df)):
+    for cell_id, t in tqdm(group_df, total=group_df.ngroups):
         res.append({"cell_id": cell_id, "length": len(t), "geom": generate_boundary(t, x=x, y=y)})
 
     return gpd.GeoDataFrame(