From d53f10c24ac9d2874842e8d38b3422bbc7cdaae6 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Tue, 24 Aug 2021 12:12:04 +0200 Subject: support override functions in dtree code --- lib/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'lib/model.py') diff --git a/lib/model.py b/lib/model.py index 76444b0..12428e3 100644 --- a/lib/model.py +++ b/lib/model.py @@ -234,7 +234,9 @@ class AnalyticModel: for name in self.names: tree_required[name] = dict() for attr in self.attr_by_name[name].keys(): - if self.attr_by_name[name][ + if self.attr_by_name[name][attr].function_override is not None: + self.attr_by_name[name][attr].fit_override_function() + elif self.attr_by_name[name][ attr ].all_relevant_parameters_are_none_or_numeric(): for key, param, args, kwargs in self.attr_by_name[name][ -- cgit v1.2.3