diff options
Diffstat (limited to 'lib/automata.py')
| -rwxr-xr-x | lib/automata.py | 5 |
1 files changed, 4 insertions, 1 deletions
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 |