From 6ae7ed3fa7be67e11c331ef5693e11aaf5ba5f81 Mon Sep 17 00:00:00 2001 From: Zane Durante Date: Thu, 25 Jul 2019 20:34:54 -0700 Subject: [PATCH] addressed comments --- sdk/python/kfp/compiler/_component_builder.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sdk/python/kfp/compiler/_component_builder.py b/sdk/python/kfp/compiler/_component_builder.py index b1bd63f214f1..d176bdb30aca 100644 --- a/sdk/python/kfp/compiler/_component_builder.py +++ b/sdk/python/kfp/compiler/_component_builder.py @@ -339,9 +339,11 @@ def _generate_entrypoint(self, component_func, python_version='python3'): func_signature = 'def ' + new_func_name + '(' for input_arg in input_args: func_signature += input_arg + ',' - func_signature += '_output_file):' if output_is_named_tuple: - func_signature = func_signature.replace('file', 'files') + func_signature += '_output_files' + else: + func_signature += '_output_file' + func_signature += '):' codegen.writeline(func_signature) # Call user function