[ML][Pipeline] Upgrade schedule operation version (#29127)

* Upgrade schedule operation version * Add test with output Signed-off-by: Brynn Yin <[email protected]> --------- Signed-off-by: Brynn Yin <[email protected]>
Azure · Mar 3, 2023 · f4fab8e · f4fab8e
1 parent c7fee36
commit f4fab8e
Show file tree

Hide file tree

Showing 12 changed files with 6,118 additions and 5,062 deletions.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
@@ -477,7 +477,7 @@ def __init__(
         self._schedules = ScheduleOperations(
             self._operation_scope,
             self._operation_config,
-            self._service_client_10_2022_preview,
+            self._service_client_12_2022_preview,
             self._operation_container,
             self._credential,
             _service_client_kwargs=kwargs,

diff --git a/...s/recordings/schedule/e2etests/test_schedule.pyTestScheduletest_command_job_schedule.json b/...s/recordings/schedule/e2etests/test_schedule.pyTestScheduletest_command_job_schedule.json
diff --git a/...etests/test_schedule.pyTestScheduletest_create_schedule_pipeline_with_output_binding.json b/...etests/test_schedule.pyTestScheduletest_create_schedule_pipeline_with_output_binding.json
diff --git a/...gs/schedule/e2etests/test_schedule.pyTestScheduletest_load_cron_schedule_with_arm_id.json b/...gs/schedule/e2etests/test_schedule.pyTestScheduletest_load_cron_schedule_with_arm_id.json
diff --git a/...e2etests/test_schedule.pyTestScheduletest_load_cron_schedule_with_arm_id_and_updates.json b/...e2etests/test_schedule.pyTestScheduletest_load_cron_schedule_with_arm_id_and_updates.json
diff --git a/...hedule/e2etests/test_schedule.pyTestScheduletest_load_cron_schedule_with_job_updates.json b/...hedule/e2etests/test_schedule.pyTestScheduletest_load_cron_schedule_with_job_updates.json
diff --git a/...hedule/e2etests/test_schedule.pyTestScheduletest_load_recurrence_schedule_no_pattern.json b/...hedule/e2etests/test_schedule.pyTestScheduletest_load_recurrence_schedule_no_pattern.json
diff --git a/...dule/e2etests/test_schedule.pyTestScheduletest_load_recurrence_schedule_with_pattern.json b/...dule/e2etests/test_schedule.pyTestScheduletest_load_recurrence_schedule_with_pattern.json
diff --git a/...ests/recordings/schedule/e2etests/test_schedule.pyTestScheduletest_schedule_lifetime.json b/...ests/recordings/schedule/e2etests/test_schedule.pyTestScheduletest_schedule_lifetime.json
diff --git a/...sts/recordings/schedule/e2etests/test_schedule.pyTestScheduletest_spark_job_schedule.json b/...sts/recordings/schedule/e2etests/test_schedule.pyTestScheduletest_spark_job_schedule.json
diff --git a/sdk/ml/azure-ai-ml/tests/schedule/e2etests/test_schedule.py b/sdk/ml/azure-ai-ml/tests/schedule/e2etests/test_schedule.py
@@ -159,6 +159,29 @@ def test_load_recurrence_schedule_with_pattern(self, client: MLClient, randstr:
             "schedule": {"hours": [10], "minutes": [15], "week_days": ["Monday"]},
         }
 
+    def test_create_schedule_pipeline_with_output_binding(self, client: MLClient, randstr: Callable[[], str]):
+        params_override = [{"name": randstr("name")}]
+        test_path = "./tests/test_configs/schedule/schedule_pipeline_with_output_binding.yml"
+        schedule = load_schedule(test_path, params_override=params_override)
+        rest_schedule = client.schedules.begin_create_or_update(schedule).result(
+            timeout=LROConfigurations.POLLING_TIMEOUT
+        )
+        assert rest_schedule.name == schedule.name
+        rest_job = rest_schedule.create_job._to_dict()
+        assert rest_job["inputs"] == {
+            "int_param": "10",
+            "data_input": {
+                "mode": "ro_mount",
+                "type": "uri_file",
+                "path": "azureml:https://dprepdata.blob.core.windows.net/demo/Titanic.csv",
+            },
+        }
+        assert rest_job["outputs"] == {
+            "output1": {"mode": "rw_mount", "type": "uri_folder"},
+            "output2": {"mode": "rw_mount", "type": "uri_folder"},
+            "output3": {"mode": "rw_mount", "type": "uri_folder"},
+        }
+
     @pytest.mark.usefixtures(
         "enable_pipeline_private_preview_features",
     )

diff --git a/sdk/ml/azure-ai-ml/tests/test_configs/schedule/schedule_pipeline_with_output_binding.yml b/sdk/ml/azure-ai-ml/tests/test_configs/schedule/schedule_pipeline_with_output_binding.yml
@@ -0,0 +1,12 @@
+$schema:  http://azureml/sdk-2-0/Schedule.json
+name: schedule_pipeline_with_output_binding
+display_name: Schedule pipeline with output binding
+description: Schedule pipeline with output binding
+
+trigger:
+  type: cron
+  expression: "15 10 * * 1"
+  time_zone: "Pacific Standard Time" # optional - default will be UTC
+
+create_job:
+  job: ../pipeline_jobs/pipeline_job_with_pipeline_component.yml