Merge branch 'main' into data/update_pancreas_dataset

theislab · Oct 11, 2024 · 59d4a5b · 59d4a5b
2 parents 20d1c4a + f91b455
commit 59d4a5b
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: