From 9b4c2868bd2e1b448aec29c3f5a90abf909d28dc Mon Sep 17 00:00:00 2001 From: Zhichang Yu Date: Wed, 20 Nov 2024 13:52:28 +0800 Subject: [PATCH] Fix set_output type hint (#3516) ### What problem does this PR solve? Fix set_output type hint ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- agent/component/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agent/component/base.py b/agent/component/base.py index 5624b7bff..62b1acdad 100644 --- a/agent/component/base.py +++ b/agent/component/base.py @@ -440,7 +440,7 @@ class ComponentBase(ABC): setattr(self._param, self._param.output_var_name, None) self._param.inputs = [] - def set_output(self, v: pd.DataFrame): + def set_output(self, v: partial | pd.DataFrame): setattr(self._param, self._param.output_var_name, v) def get_input(self):