From 573afacd2b94a42a988f4c16bff4664bdf14595f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 28 Oct 2024 09:12:18 -0300 Subject: [PATCH] Update test to use _outputs_map for cycle outputs retrieval --- src/backend/tests/unit/graph/graph/test_cycles.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/tests/unit/graph/graph/test_cycles.py b/src/backend/tests/unit/graph/graph/test_cycles.py index fe764be4f38..44537cda998 100644 --- 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