From 80319e7188f14f167c5c3dd0158408160ced2b93 Mon Sep 17 00:00:00 2001
From: Yaqi <jiyaqi0712@gmail.com>
Date: Fri, 15 Oct 2021 14:09:51 -0700
Subject: [PATCH] edit

---
 google/cloud/aiplatform/pipeline_jobs.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/google/cloud/aiplatform/pipeline_jobs.py b/google/cloud/aiplatform/pipeline_jobs.py
index 389a6171aa..395e4102d3 100644
--- 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()