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
|