diff options
| -rwxr-xr-x | lib/dfatool.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/lib/dfatool.py b/lib/dfatool.py index 1802acf..8f2e9e6 100755 --- a/lib/dfatool.py +++ b/lib/dfatool.py @@ -1075,6 +1075,8 @@ class AnalyticModel: } def model_getter(name, key, **kwargs): + if 'arg' in kwargs and 'param' in kwargs: + kwargs['param'].extend(map(soft_cast_int, kwargs['arg'])) if key in param_model[name]: param_list = kwargs['param'] param_function = param_model[name][key]['function'] @@ -1450,6 +1452,8 @@ class PTAModel: } def model_getter(name, key, **kwargs): + if 'arg' in kwargs and 'param' in kwargs: + kwargs['param'].extend(map(soft_cast_int, kwargs['arg'])) if key in param_model[name]: param_list = kwargs['param'] param_function = param_model[name][key]['function'] |