Fix sankey threshold (#754)

Co-authored-by: Leon Stadelmann <leon.stadelmann@helmholtz-munich.de>
theislab · Oct 11, 2024 · f91b455 · f91b455
1 parent ba8d30f
commit f91b455
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/moscot/problems/time/_mixins.py b/src/moscot/problems/time/_mixins.py
@@ -379,7 +379,7 @@ def sankey(
             cell_transitions_updated = cell_transitions
 
         if threshold is not None:
-            for ct in cell_transitions:
+            for ct in cell_transitions_updated:
                 ct[ct < threshold] = 0.0
 
         if key_added is None: