Update src/pixelator/pixeldataset/utils.py

Co-authored-by: Johan Dahlberg <[email protected]>
PixelgenTechnologies · Oct 22, 2024 · 13f6618 · 13f6618
1 parent 4d38f39
commit 13f6618
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/pixelator/pixeldataset/utils.py b/src/pixelator/pixeldataset/utils.py
@@ -221,6 +221,8 @@ def write_anndata(adata: AnnData, filename: PathType) -> None:
 
 def _compute_sub_communities(
     component_edgelist: pd.DataFrame, n_edges_reconnect: int | None = None
+) -> pd.Series:
+    component_edgelist: pd.DataFrame, n_edges_reconnect: int | None = None
 ):
     component_edgelist = (
         component_edgelist.groupby(["upia", "upib"], observed=True)["count"]