From 146205a761a62f999a371a1d25cc1f91665f9257 Mon Sep 17 00:00:00 2001 From: Jeremy Zucker Date: Tue, 1 Oct 2024 15:09:20 -0700 Subject: [PATCH] Update util.py Added type check for int, float, or str and raise a value error if none of the above --- mira/sources/util.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mira/sources/util.py b/mira/sources/util.py index 414ed349..fa780243 100644 --- a/mira/sources/util.py +++ b/mira/sources/util.py @@ -188,9 +188,11 @@ def parameter_to_mira(parameter, param_symbols=None) -> Parameter: for param_key, param_value in distr_json.get("parameters", {}).items(): if isinstance(param_value, float) or isinstance(param_value, int): processed_distr_parameters[param_key] = param_value - else: + elif isinstance(param_value, str): processed_distr_parameters[param_key] = \ safe_parse_expr(param_value) + else: + raise ValueError(f"{param_value} is neither a float, int, or str") distr = Distribution( type=distr_type, parameters=processed_distr_parameters,