From b7cb39b41de322fad27a9b1f5893266c2a6f1371 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 14:35:45 -0300 Subject: [PATCH] Fix assertion in test_build_graph --- tests/test_cache.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_cache.py b/tests/test_cache.py index 925402769..c2c706ee9 100644 --- a/tests/test_cache.py +++ b/tests/test_cache.py @@ -1,7 +1,7 @@ import json +from langflow.graph import Graph import pytest -from langflow.graph import Graph def get_graph(_type="basic"): @@ -41,5 +41,5 @@ def langchain_objects_are_equal(obj1, obj2): def test_build_graph(client, basic_data_graph): graph = Graph.from_payload(basic_data_graph) assert graph is not None - assert len(graph.vertices) == len(basic_data_graph["nodes"]) + assert len(graph.nodes) == len(basic_data_graph["nodes"]) assert len(graph.edges) == len(basic_data_graph["edges"])