Skip to content

Commit f7e8fcd

Browse files
committed
Update graph visualization util to work in Instruqt
1 parent 5da0df4 commit f7e8fcd

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

notebooks/utils/utils.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,5 +71,14 @@ def check_index_ready(collection: Collection, index_name: str) -> None:
7171
print(f"{index_name} index status: {status}")
7272
time.sleep(SLEEP_TIMER)
7373

74-
def visualize_graph(graph):
75-
display(Image(graph.draw_mermaid_png()))
74+
def visualize_graph(app):
75+
"""
76+
Visualize the graph of the application using Mermaid.
77+
78+
Args:
79+
app: The application instance containing the graph to visualize.
80+
81+
Returns:
82+
None
83+
"""
84+
display(Image(app.get_graph().draw_mermaid_png()))

0 commit comments

Comments
 (0)