Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
@@ -106,7 +106,7 @@ class Visualization:
|
|
106 |
|
107 |
return dot
|
108 |
|
109 |
-
def download_tree_diagram(dot, filename):
|
110 |
# Render the diagram to a file
|
111 |
filepath = dot.render(filename, format='png', cleanup=True)
|
112 |
return filepath
|
|
|
106 |
|
107 |
return dot
|
108 |
|
109 |
+
def download_tree_diagram(self, dot, filename):
|
110 |
# Render the diagram to a file
|
111 |
filepath = dot.render(filename, format='png', cleanup=True)
|
112 |
return filepath
|