From 498f986de757325e4aafc990d3faa2925d236df7 Mon Sep 17 00:00:00 2001 From: Justus Kuhlmann Date: Tue, 2 Sep 2025 10:25:49 +0000 Subject: [PATCH] better replacements --- corrlib/toml.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/corrlib/toml.py b/corrlib/toml.py index def5394..abb5545 100644 --- a/corrlib/toml.py +++ b/corrlib/toml.py @@ -29,12 +29,11 @@ def replace_in_meas(measurements: dict, vars: dict[str, str]): for name, value in vars.items(): for m in measurements.keys(): for key in measurements[m].keys(): - measurements[m][key] = replace_string(measurements[m][key], name, value) - # replace ensemble name - for m in measurements.keys(): - for key in measurements[m].keys(): - if not key == 'ensemble': - measurements[m][key] = replace_string(measurements[m][key], 'ensemble', measurements[m]['ensemble']) + if isinstance(measurements[m][key], str): + measurements[m][key] = replace_string(measurements[m][key], name, value) + elif isinstance(measurements[m][key], list): + for i in range(len(measurements[m][key])): + measurements[m][key][i] = replace_string(measurements[m][key][i], name, value) return measurements def fill_cons(measurements, constants):