SDK - Components - Fixed bug in loading input-less graph components (k…

…ubeflow#3446)
Jeffwan · Dec 9, 2020 · 0eb3fa5 · 0eb3fa5
1 parent 2762e11
commit 0eb3fa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/kfp/components/_components.py b/sdk/python/kfp/components/_components.py
@@ -484,7 +484,7 @@ def _resolve_graph_task(
 
     graph = component_spec.implementation.graph
 
-    graph_input_arguments = {input.name: input.default for input in component_spec.inputs if input.default is not None}
+    graph_input_arguments = {input.name: input.default for input in component_spec.inputs or [] if input.default is not None}
     graph_input_arguments.update(arguments)
 
     outputs_of_tasks = {}