From f1a1f584f776798d5ee5ea3f16d517d1198ca63e Mon Sep 17 00:00:00 2001 From: Leon Stadelmann Date: Thu, 10 Oct 2024 19:31:30 +0200 Subject: [PATCH] Fix sankey threshold --- 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 29a44aed0..7d030d6ec 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: