From 7989d4fdace6e46f31ccb45f027ba91ae34bc8e8 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 29 Mar 2022 10:11:53 +0200
Subject: parameters: to_dot: Allow exporting single-node trees

---
 lib/parameters.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/parameters.py b/lib/parameters.py
index cb4b76f..01b3e66 100644
--- a/lib/parameters.py
+++ b/lib/parameters.py
@@ -704,7 +704,11 @@ class ModelAttribute:
         return ret
 
     def to_dot(self):
-        if type(self.model_function) == df.SplitFunction:
+        if type(self.model_function) in (
+            df.SplitFunction,
+            df.StaticFunction,
+            df.AnalyticFunction,
+        ):
             import pydot
 
             graph = pydot.Dot("Regression Model Tree", graph_type="graph")
-- 
cgit v1.2.3