fix: Typechek

PixelgenTechnologies · Oct 22, 2024 · 37d1ddf · 37d1ddf
1 parent dde6fbe
commit 37d1ddf
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/pixelator/graph/community_detection.py b/src/pixelator/graph/community_detection.py
@@ -189,14 +189,14 @@ def connect_components(
     result_metrics = edgelist_metrics(graph_output_edgelist)
 
     result_metrics["edges_with_colliding_upi_count"] = (
-        removed_edgelist_df["depth"] == 0
-    ).sum()
+        (removed_edgelist_df["depth"] == 0).sum().astype(int)
+    )
     result_metrics["edges_removed_in_multiplet_recovery_first_iteration"] = (
-        removed_edgelist_df["depth"] == 1
-    ).sum()
+        (removed_edgelist_df["depth"] == 1).sum().astype(int)
+    )
     result_metrics["edges_removed_in_multiplet_recovery_refinement"] = (
-        removed_edgelist_df["depth"] > 1
-    ).sum()
+        (removed_edgelist_df["depth"] > 1).sum().astype(int)
+    )
     result_metrics["fraction_edges_removed_in_refinement"] = (
         removed_edgelist_df["depth"] > 1
     ).sum() / max(len(removed_edgelist_df), 1)