From 5c352157d1b13bfc69041eece40df322eebc1fe8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 4 Dec 2023 12:56:10 -0300 Subject: [PATCH] Fix return type assertion in test_custom_component_get_function_entrypoint_return_type() --- tests/test_custom_component.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_custom_component.py b/tests/test_custom_component.py index 636bd63b1..64165fa9a 100644 --- a/tests/test_custom_component.py +++ b/tests/test_custom_component.py @@ -4,7 +4,7 @@ from uuid import uuid4 import pytest from fastapi import HTTPException -from langflow.field_typing.constants import Data + from langflow.interface.custom.base import CustomComponent from langflow.interface.custom.code_parser import CodeParser, CodeSyntaxError from langflow.interface.custom.component import Component, ComponentCodeNullError @@ -412,7 +412,7 @@ class MyClass(CustomComponent): custom_component = CustomComponent(code=my_code, function_entrypoint_name="build") return_type = custom_component.get_function_entrypoint_return_type - assert return_type == [Data] + assert return_type == [] def test_custom_component_get_main_class_name_no_main_class():