SDK - Lightweight - Fixed regression for components without outputs (#…

…1726)
kubeflow · Aug 6, 2019 · dd59bc2 · dd59bc2
1 parent 4fb329e
commit dd59bc2
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/sdk/python/kfp/components/_python_op.py b/sdk/python/kfp/components/_python_op.py
@@ -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 = \
 '''\