From 77e112278262c960bfccfce353fb035f2139c85b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:56:49 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20(component.py):=20fix=20type=20h?= =?UTF-8?q?int=20for=20map=5Finputs=20method=20parameter=20to=20match=20th?= =?UTF-8?q?e=20imported=20InputTypes=20class=20for=20better=20clarity=20an?= =?UTF-8?q?d=20consistency?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../base/langflow/custom/custom_component/component.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py index d75316218..4d4bdc225 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -9,7 +9,7 @@ from langflow.inputs.inputs import InputTypes from langflow.schema.artifact import get_artifact_type, post_process_raw from langflow.schema.data import Data from langflow.schema.message import Message -from langflow.template.field.base import UNDEFINED, Input, Output +from langflow.template.field.base import UNDEFINED, Output from .custom_component import CustomComponent @@ -63,7 +63,7 @@ class Component(CustomComponent): # except AttributeError: # return self.__getattr__(name) - def map_inputs(self, inputs: List[Input]): + def map_inputs(self, inputs: List[InputTypes]): self.inputs = inputs for input_ in inputs: self._inputs[input_.name] = input_