jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 (component.py): fix type hint for map_inputs method parameter to match the imported InputTypes class for better clarity and consistency

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:56:49 -03:00
commit 77e1122782
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/custom/custom_component/component.py
View file

@ -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_