edit

googleapis · Oct 15, 2021 · 80319e7 · 80319e7
1 parent 47cf534
commit 80319e7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/google/cloud/aiplatform/pipeline_jobs.py b/google/cloud/aiplatform/pipeline_jobs.py
@@ -204,9 +204,10 @@ def __init__(
                 pipeline_root=pipeline_root,
                 parameter_types={
                     key: value["type"]
-                    for key, value in pipeline_job["pipelineSpec"]["root"][
-                        "inputDefinitions"
-                    ]["parameters"].items()
+                    for key, value in pipeline_job["pipelineSpec"]["root"]
+                    .get("inputDefinitions", {})
+                    .get("parameters", {})
+                    .items()
                 },
                 parameter_values=parameter_values,
             ).build()