Update visualization.py

visualization.py

@@ -73,7 +73,7 @@ class Visualization:
         st.pyplot(fig)
         return fig
 
-    def build_tree_diagram(row):
+    def build_tree_diagram(self, row):
         dot = Digraph()
     
         # Add the student's name as the first node