From f91b4559c4c95baae1d89fd4eb0db36a84cc1f3f Mon Sep 17 00:00:00 2001 From: LeonStadelmann <169152764+LeonStadelmann@users.noreply.github.com> Date: Fri, 11 Oct 2024 16:52:03 +0200 Subject: [PATCH] Fix sankey threshold (#754) Co-authored-by: Leon Stadelmann --- src/moscot/problems/time/_mixins.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/moscot/problems/time/_mixins.py b/src/moscot/problems/time/_mixins.py index 4673934c1..0fde8a334 100644 --- 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: