From 5b388abe294a35174abe83f0987e0d41006fc79f Mon Sep 17 00:00:00 2001 From: zjgemi Date: Thu, 5 Dec 2024 20:22:32 +0800 Subject: [PATCH] fix: set output parameter slices Signed-off-by: zjgemi --- src/dflow/python/python_op_template.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/dflow/python/python_op_template.py b/src/dflow/python/python_op_template.py index d3fca265..951195e6 100644 --- a/src/dflow/python/python_op_template.py +++ b/src/dflow/python/python_op_template.py @@ -601,6 +601,9 @@ def render_script(self): if isinstance(sign, Artifact): slices = self.get_slices(output_artifact_slices, name) script += " op_obj.slices['%s'] = %s\n" % (name, slices) + else: + slices = self.get_slices(output_parameter_slices, name) + script += " op_obj.slices['%s'] = %s\n" % (name, slices) script += " import signal\n" script += " def sigterm_handler(signum, frame):\n"