Fix

langchain-ai · Dec 5, 2024 · 9f73dfa · 9f73dfa
1 parent 4459952
commit 9f73dfa
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/libs/langgraph/langgraph/graph/graph.py b/libs/langgraph/langgraph/graph/graph.py
@@ -630,10 +630,9 @@ def add_edge(
 
         return graph
 
-    def _repr_mimebundle_(self, **kwargs) -> dict[str, Any]:
+    def _repr_mimebundle_(self, **kwargs: Any) -> dict[str, Any]:
         """Mime bundle used by Jupyter to display the graph"""
-        output = {
+        return {
             "text/plain": repr(self),
             "image/png": self.get_graph().draw_mermaid_png(),
         }
-        return output