edit

googleapis · Oct 15, 2021 · a31e745 · a31e745
1 parent 47cf534
commit a31e745
Showing 1 changed file with 17 additions and 10 deletions.
diff --git a/google/cloud/aiplatform/pipeline_jobs.py b/google/cloud/aiplatform/pipeline_jobs.py
@@ -200,16 +200,23 @@ def __init__(
             )
 
         if pipeline_spec_only:
-            runtime_config_dict = pipeline_utils.PipelineRuntimeConfigBuilder(
-                pipeline_root=pipeline_root,
-                parameter_types={
-                    key: value["type"]
-                    for key, value in pipeline_job["pipelineSpec"]["root"][
-                        "inputDefinitions"
-                    ]["parameters"].items()
-                },
-                parameter_values=parameter_values,
-            ).build()
+            try:
+                runtime_config_dict = pipeline_utils.PipelineRuntimeConfigBuilder(
+                    pipeline_root=pipeline_root,
+                    parameter_types={
+                        key: value["type"]
+                        for key, value in pipeline_job["pipelineSpec"]["root"][
+                            "inputDefinitions"
+                        ]["parameters"].items()
+                    },
+                    parameter_values=parameter_values,
+                ).build()
+            except AttributeError:
+                runtime_config_dict = pipeline_utils.PipelineRuntimeConfigBuilder(
+                    pipeline_root=pipeline_root,
+                    parameter_types={},
+                    parameter_values=parameter_values,
+                ).build()
         else:
             builder = pipeline_utils.PipelineRuntimeConfigBuilder.from_job_spec_json(
                 pipeline_job