From 77c9eac2a5ce2d42a56924e8423a86cfb88472a7 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 15:57:58 -0300 Subject: [PATCH] refactor: Initialize Component class with empty dictionaries for inputs, results, and arguments --- .../base/langflow/custom/custom_component/component.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py index 9829e695f..58a387f1b 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -42,17 +42,16 @@ class Component(CustomComponent): inputs: Optional[List[InputTypes]] = None outputs: Optional[List[Output]] = None code_class_base_inheritance: ClassVar[str] = "Component" - _results: dict = {} - _arguments: dict = {} - _inputs: dict[str, InputTypes] = {} def __init__(self, **data): super().__init__(**data) + self._inputs: dict[str, InputTypes] = {} + self._results: dict = {} + self._arguments: dict = {} if self.inputs is not None: self.map_inputs(self.inputs) def map_inputs(self, inputs: List[Input]): - self._inputs = {} self.inputs = inputs for input_ in inputs: self._inputs[input_.name] = input_