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"