fix: Support CMEK for scheduled pipeline jobs.

vertex-sdk-bot · copybara-github · commit 1c90c88b7410 · 2023-12-12T14:32:49.000-08:00
FUTURE_COPYBARA_INTEGRATE_REVIEW=#3066 from googleapis:release-please--branches--main 5353beb PiperOrigin-RevId: 590338288
diff --git a/google/cloud/aiplatform/pipeline_job_schedules.py b/google/cloud/aiplatform/pipeline_job_schedules.py
@@ -106,6 +106,10 @@ def __init__(
             create_pipeline_job_request["pipeline_job"][
                 "labels"
             ] = pipeline_job._gca_resource.labels
+        if "encryption_spec" in pipeline_job._gca_resource:
+            create_pipeline_job_request["pipeline_job"][
+                "encryption_spec"
+            ] = pipeline_job._gca_resource.encryption_spec
         pipeline_job_schedule_args = {
             "display_name": display_name,
             "create_pipeline_job_request": create_pipeline_job_request,
