From 503437c73a88a006a8d452482ef13c1533fbf61a Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 19:54:32 -0300 Subject: [PATCH] refactor: Update CustomComponent to use data_objects instead of data list --- .../langflow/custom/custom_component/custom_component.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/custom/custom_component/custom_component.py b/src/backend/base/langflow/custom/custom_component/custom_component.py index 9eec57693..92631c140 100644 --- a/src/backend/base/langflow/custom/custom_component/custom_component.py +++ b/src/backend/base/langflow/custom/custom_component/custom_component.py @@ -219,7 +219,7 @@ class CustomComponent(BaseComponent): """ if not keys: keys = [] - data = [] + data_objects = [] if not isinstance(data, Sequence): data = [data] for item in data: @@ -245,9 +245,9 @@ class CustomComponent(BaseComponent): else: raise ValueError(f"Invalid data type: {type(item)}") - data.append(Data(data=data_dict)) + data_objects.append(Data(data=data_dict)) - return data + return data_objects def create_references_from_data(self, data: List[Data], include_data: bool = False) -> str: """