From 4e8e51126d035d009cd962e3639848a01075f0f4 Mon Sep 17 00:00:00 2001
From: Angel Igareta <angel.igareta@klarna.com>
Date: Wed, 17 Apr 2024 18:03:22 +0200
Subject: [PATCH] Fix formatting in graph

---
 langgraph/graph/graph.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py
index 0d4a8a093..0be33a53b 100644
--- a/langgraph/graph/graph.py
+++ b/langgraph/graph/graph.py
@@ -364,7 +364,11 @@ class CompiledGraph(Pregel):
                 self.nodes[end].channels.append(channel_name)
 
     def get_graph(
-        self, config: Optional[RunnableConfig] = None, *, xray: bool = False, add_condition_nodes: bool = True
+        self,
+        config: Optional[RunnableConfig] = None,
+        *,
+        xray: bool = False,
+        add_condition_nodes: bool = True,
     ) -> RunnableGraph:
         """Returns a drawable representation of the computation graph."""
         graph = RunnableGraph()
@@ -405,7 +409,7 @@ class CompiledGraph(Pregel):
                     END: END,
                 }
 
-                if add_condition_nodes is True: 
+                if add_condition_nodes is True:
                     cond = graph.add_node(branch.condition, name)
                     graph.add_edge(start_nodes[start], cond)
                     for label, end in ends.items():