From 0886a5fe14195fc105939b477f9ff70b27ae3172 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Tue, 31 Aug 2021 14:59:14 +0200 Subject: model: allow leaf function setting to be overridden via env Here, it defaults to enabled and can be overridden to disabled (DFATOOL_DTREE_FUNCTION_LEAVES=0) --- lib/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/model.py b/lib/model.py index 696813e..3f784cc 100644 --- a/lib/model.py +++ b/lib/model.py @@ -255,11 +255,14 @@ class AnalyticModel: for name in self.names: for attr in self.attr_by_name[name].keys(): if tree_required[name].get(attr, False): + with_function_leaves = True + if os.getenv("DFATOOL_DTREE_FUNCTION_LEAVES") == "0": + with_function_leaves = False self.build_dtree( name, attr, self.attr_by_name[name][attr].stats.std_param_lut, - with_function_leaves=False, + with_function_leaves=with_function_leaves, ) else: self.attr_by_name[name][attr].set_data_from_paramfit(paramfit) -- cgit v1.2.3