ProfessorLeVesseur commited on
Commit
cd77695
·
verified ·
1 Parent(s): bed569d

Update visualization.py

Browse files
Files changed (1) hide show
  1. 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