From 70f3c354cbeea4123917badc60f3d0f184b08c77 Mon Sep 17 00:00:00 2001 From: Daniel Weindl Date: Tue, 2 Jan 2024 12:37:01 +0100 Subject: [PATCH] Apply suggestions from code review MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Fabian Fröhlich --- python/sdist/amici/plotting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sdist/amici/plotting.py b/python/sdist/amici/plotting.py index 2826f2c3ea..edf4a33156 100644 --- a/python/sdist/amici/plotting.py +++ b/python/sdist/amici/plotting.py @@ -56,7 +56,7 @@ def plot_state_trajectories( elif model is not None and prefer_names: labels = np.asarray(model.getStateNames())[list(state_indices)] labels = [ - l if l else model.getStateNames()[ix] + l if l else model.getStateIds()[ix] for ix, l in enumerate(labels) ] elif model is not None: @@ -116,7 +116,7 @@ def plot_observable_trajectories( list(observable_indices) ] labels = [ - l if l else model.getObservableNames()[ix] + l if l else model.getObservableIds()[ix] for ix, l in enumerate(labels) ] elif model is not None: