Update src/pixelator/pixeldataset/utils.py

Co-authored-by: Johan Dahlberg <johan.dahlberg@pixelgen.tech>
PixelgenTechnologies · Oct 22, 2024 · a427c36 · a427c36
1 parent 13f6618
commit a427c36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pixelator/pixeldataset/utils.py b/src/pixelator/pixeldataset/utils.py
@@ -243,7 +243,7 @@ def _compute_sub_communities(
     return component_communities
 
 
-def _assess_doublet(component_edgelist: pd.DataFrame):
+def _assess_doublet(component_edgelist: pd.DataFrame) -> tuple[bool, int]:
     """Check whether a component is a potential doublet and how many edges should be removed to split it.
 
     A component is a potential doublet if a) it has more than one community and