fix: do not mark caller vertex

langflow-ai · Aug 14, 2024 · 9de1f56 · 9de1f56
1 parent 899a62a
commit 9de1f56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/backend/base/langflow/graph/graph/base.py b/src/backend/base/langflow/graph/graph/base.py
@@ -788,12 +788,12 @@ def _mark_branch(
         """Marks a branch of the graph."""
         if visited is None:
             visited = set()
+        else:
+            self.mark_vertex(vertex_id, state)
         if vertex_id in visited:
             return
         visited.add(vertex_id)
 
-        self.mark_vertex(vertex_id, state)
-
         for child_id in self.parent_child_map[vertex_id]:
             # Only child_id that have an edge with the vertex_id through the output_name
             # should be marked