SDK - Lightweight - Fixed regression for components without outputs (#1726)

This commit is contained in:
Alexey Volkov 2019-08-05 21:47:53 -07:00 committed by Kubernetes Prow Robot
parent 4fb329eaf8
commit dd59bc2597
1 changed files with 3 additions and 4 deletions

View File

@ -287,10 +287,9 @@ def _func_to_component_spec(func, extra_code='', base_image=_default_base_image,
'_parsed_args = vars(_parser.parse_args())',
])
if component_spec.outputs:
arg_parse_code_lines.extend([
'_output_files = _parsed_args.pop("_output_paths")',
])
arg_parse_code_lines.extend([
'_output_files = _parsed_args.pop("_output_paths", [])',
])
full_source = \
'''\