fix: Fix artifact registry link not showing in ui when creating schedules with SDK.

PiperOrigin-RevId: 547572327

Author: vertex-sdk-bot

google/cloud/aiplatform/preview/pipelinejobschedule/pipeline_job_schedules.py

@@ -107,6 +107,10 @@ def __init__(
                 "pipeline_spec": pipeline_job.pipeline_spec,
             },
         }
+        if "template_uri" in pipeline_job._gca_resource:
+            create_pipeline_job_request["pipeline_job"][
+                "template_uri"
+            ] = pipeline_job._gca_resource.template_uri
         pipeline_job_schedule_args = {
             "display_name": display_name,
             "create_pipeline_job_request": create_pipeline_job_request,

tests/unit/aiplatform/test_pipeline_job_schedules.py

@@ -658,6 +658,7 @@ def test_call_schedule_service_create_artifact_registry(
                     "pipeline_spec": dict_to_struct(pipeline_spec),
                     "service_account": _TEST_SERVICE_ACCOUNT,
                     "network": _TEST_NETWORK,
+                    "template_uri": _TEST_AR_TEMPLATE_PATH,
                 },
             },
         )
@@ -740,6 +741,7 @@ def test_call_schedule_service_create_https(
                     "pipeline_spec": dict_to_struct(pipeline_spec),
                     "service_account": _TEST_SERVICE_ACCOUNT,
                     "network": _TEST_NETWORK,
+                    "template_uri": _TEST_HTTPS_TEMPLATE_PATH,
                 },
             },
         )