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)

1 files changed, 4 insertions, 1 deletions

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)