From 7275dc42b5358bde2be9cccb9b079b21874a98b2 Mon Sep 17 00:00:00 2001 From: Didier Vezinet Date: Mon, 27 Jan 2025 12:58:44 -0500 Subject: [PATCH] [#1017] Implemented --- tofu/data/_class01_load_equilibrium.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tofu/data/_class01_load_equilibrium.py b/tofu/data/_class01_load_equilibrium.py index 12f45375a..c344b37d9 100644 --- a/tofu/data/_class01_load_equilibrium.py +++ b/tofu/data/_class01_load_equilibrium.py @@ -197,8 +197,13 @@ def main( if sort_vs is not None: - lok = [kk for kk in ddata.keys() if len(ddata[kk]['ref']) == 1] - if sort_vs in lok: + lok_att = [kk for kk in ddata.keys() if len(ddata[kk]['ref']) == 1] + lok_key = [ddata[kk]['key'] for kk in lok_att] + if sort_vs in lok_key: + sort_vs = [ + kk for kk in lok_att if ddata[kk]['key'] == sort_vs + ][0] + if sort_vs in lok_att: inds = np.argsort(ddata[sort_vs]['data']) for kk, vv in ddata.items(): sli = (inds,) + tuple([slice(None) for ss in vv['data'].shape[1:]])