dref export: static and analytic function ≙ single-node tree

1 files changed, 3 insertions, 0 deletions

diff --git a/lib/parameters.py b/lib/parameters.py
index faa2e96..8e17f2b 100644
--- a/lib/parameters.py
+++ b/lib/parameters.py
@@ -644,6 +644,9 @@ class ModelAttribute:
         ):
             ret["decision tree/nodes"] = self.model_function.get_number_of_nodes()
             ret["decision tree/max depth"] = self.model_function.get_max_depth()
+        elif type(self.model_function) in (df.StaticFunction, df.AnalyticFunction):
+            ret["decision tree/nodes"] = 1
+            ret["decision tree/max depth"] = 1
 
         return ret