From b0e65124da0330713cd72c42254c153974b710eb Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Fri, 26 Jul 2019 07:47:53 +0200 Subject: add parameter normalization support to YAML DFA models --- lib/automata.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'lib/automata.py') diff --git a/lib/automata.py b/lib/automata.py index 6437839..3220d81 100755 --- a/lib/automata.py +++ b/lib/automata.py @@ -1,6 +1,6 @@ """Classes and helper functions for PTA and other automata.""" -from functions import AnalyticFunction +from functions import AnalyticFunction, NormalizationFunction import itertools def _dict_to_list(input_dict: dict) -> list: @@ -310,6 +310,9 @@ class PTA: if parameter in self.parameter_normalization: if 'enum' in self.parameter_normalization[parameter] and value in self.parameter_normalization[parameter]['enum']: normalized_param[parameter] = self.parameter_normalization[parameter]['enum'][value] + if 'formula' in self.parameter_normalization[parameter]: + normalization_formula = NormalizationFunction(self.parameter_normalization[parameter]['formula']) + normalized_param[parameter] = normalization_formula.eval(value) return normalized_param @classmethod -- cgit v1.2.3