From 281704478ff12605df8b5e16bd5c47aa14a6f1de Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Thu, 26 Sep 2019 21:03:17 -0700 Subject: [PATCH] SDK - Compiler - Fixed small bug in data passing rewriter --- sdk/python/kfp/compiler/_data_passing_rewriter.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sdk/python/kfp/compiler/_data_passing_rewriter.py b/sdk/python/kfp/compiler/_data_passing_rewriter.py index 74ff74fc549..2428a3b3a97 100644 --- a/sdk/python/kfp/compiler/_data_passing_rewriter.py +++ b/sdk/python/kfp/compiler/_data_passing_rewriter.py @@ -328,9 +328,11 @@ def mark_upstream_ios_of_output(template_output, marked_inputs, marked_outputs): for parameter_argument in parameter_arguments: input_name = parameter_argument['name'] if (entrypoint_template_name, input_name) in inputs_consumed_as_artifacts: - artifact_arguments.add({ + artifact_arguments.append({ 'name': input_name, - 'raw': '{{workflow.parameters.' + input_name + '}}' + 'raw': { + 'data': '{{workflow.parameters.' + input_name + '}}', + }, }) if artifact_arguments: workflow_arguments['artifacts'] = artifact_arguments