avoid devision by zero (#139)

dpeerlab · Apr 22, 2024 · 12e363b · 12e363b
1 parent 5801a16
commit 12e363b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/palantir/presults.py b/src/palantir/presults.py
@@ -551,7 +551,9 @@ def select_branch_cells(
     prob_thresholds = np.empty_like(fate_probs)
     n = fate_probs.shape[0]
 
-    step = n // PSEUDOTIME_RES
+    pseudotime_resolution = min(PSEUDOTIME_RES, n)
+
+    step = n // pseudotime_resolution
     nsteps = n // step
     for i in range(nsteps):
         l, r = i * step, (i + 1) * step