Update test to use _outputs_map for cycle outputs retrieval

langflow-ai · Oct 28, 2024 · 1d49939 · 1d49939
1 parent 4f5113b
commit 1d49939
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/backend/tests/unit/graph/graph/test_cycles.py b/src/backend/tests/unit/graph/graph/test_cycles.py
@@ -127,7 +127,9 @@ def test_that_outputs_cache_is_set_to_false_in_cycle():
 
     graph = Graph(chat_input, chat_output)
     cycle_vertices = find_cycle_vertices(graph._get_edges_as_list_of_tuples())
-    cycle_outputs_lists = [graph.vertex_map[vertex_id].custom_component.outputs for vertex_id in cycle_vertices]
+    cycle_outputs_lists = [
+        graph.vertex_map[vertex_id].custom_component._outputs_map.values() for vertex_id in cycle_vertices
+    ]
     cycle_outputs = [output for outputs in cycle_outputs_lists for output in outputs]
     for output in cycle_outputs:
         assert output.cache is False