From 6e041a62b8e38dd55b25fcf3ef77c607cf6542b3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 4 Dec 2023 12:09:08 -0300 Subject: [PATCH] Fix assertion in test_build_graph --- tests/test_cache.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_cache.py b/tests/test_cache.py index c2c706ee9..8cfebe230 100644 --- a/tests/test_cache.py +++ b/tests/test_cache.py @@ -1,8 +1,9 @@ import json -from langflow.graph import Graph import pytest +from langflow.graph import Graph + def get_graph(_type="basic"): """Get a graph from a json file""" @@ -41,5 +42,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.nodes) == len(basic_data_graph["nodes"]) + assert len(graph.vertices) == len(basic_data_graph["nodes"]) assert len(graph.edges) == len(basic_data_graph["edges"])