From 0313592a7b9bc34542b055561b8c05eadb1f1f98 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Fri, 19 Jan 2024 16:42:47 -0300 Subject: [PATCH] Fix graph edge references in Graph class --- src/backend/langflow/graph/graph/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py index dc5e6db5e..3fa726a37 100644 --- a/src/backend/langflow/graph/graph/base.py +++ b/src/backend/langflow/graph/graph/base.py @@ -261,8 +261,8 @@ class Graph: # Build graph and compute in-degrees for edge in self.edges: - graph[edge.source].append(edge.target) - in_degree[edge.target] += 1 + graph[edge.source_id].append(edge.target_id) + in_degree[edge.target_id] += 1 # Queue for vertices with no incoming edges queue = deque(vertex for vertex in self.vertices if in_degree[vertex] == 0)