fixed exception in tl.slide_cells

Author: LouisFaure

scFates/tools/slide_cors.py

@@ -134,7 +134,8 @@ def region_extract(pt_cur, segs_cur, nbranch):
             (graph["pp_info"].loc[pps, "time"].values >= pt_cur)
             & graph["pp_info"].loc[pps, "seg"].isin(segs_cur).values
         ]
-
+        if len(pp_next) == 0:
+            raise Exception("win parameter too small, increase number of cells")
         cmsm = np.cumsum(
             pp_probs[pp_next][np.argsort(graph["pp_info"].loc[pp_next, "time"].values)]
         )
@@ -176,6 +177,10 @@ def region_extract(pt_cur, segs_cur, nbranch):
             freq = freq + [cell_probs]
             pt_cur = graph["pp_info"].loc[pps_region, "time"].max()
 
+            # ignore the last node, to avoid infinite recursion
+            if len(pps_region) == 1:
+                return freq
+
             if nbranch == 1:
                 if (
                     sum(~graph["pp_info"].loc[pps_region, :].seg.isin(seg_progenies))