From ca4aa52f635ba46caf8927591ad13cd4bcbb1fc7 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <derf@finalrewind.org>
Date: Wed, 3 Apr 2019 11:07:05 +0200
Subject: dfatool: analytic model getter: Add arg support
---
lib/dfatool.py | 4 ++++
1 file changed, 4 insertions(+)
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']
--
cgit v1.2.3