From a41ada87ee42946bb81ffcfd1a0621c90324ebf4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 25 Jan 2024 16:17:20 -0300 Subject: [PATCH] Fix import and function name in test_loading.py --- tests/test_loading.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_loading.py b/tests/test_loading.py index 8a797579f..a1a2ec1f4 100644 --- a/tests/test_loading.py +++ b/tests/test_loading.py @@ -1,9 +1,10 @@ import json + import pytest from langchain.chains.base import Chain -from langflow.processing.process import load_flow_from_json from langflow.graph import Graph -from langflow.utils.payload import get_root_node +from langflow.processing.load import load_flow_from_json +from langflow.utils.payload import get_root_vertex def test_load_flow_from_json(): @@ -22,14 +23,14 @@ def test_load_flow_from_json_with_tweaks(): assert loaded.llm.model_name == "gpt-3.5-turbo-16k-0613" -def test_get_root_node(): +def test_get_root_vertex(): with open(pytest.BASIC_EXAMPLE_PATH, "r") as f: flow_graph = json.load(f) data_graph = flow_graph["data"] nodes = data_graph["nodes"] edges = data_graph["edges"] graph = Graph(nodes, edges) - root = get_root_node(graph) + root = get_root_vertex(graph) assert root is not None assert hasattr(root, "id") assert hasattr(root, "data")