Skip to content

Commit ebcfffd

Browse files
chnldwDawei Li
andauthored
Add e2e test for distribution:ray (Azure#30044)
Co-authored-by: Dawei Li <daweil@microsoft.com>
1 parent e73bb1e commit ebcfffd

File tree

3 files changed

+2633
-0
lines changed

3 files changed

+2633
-0
lines changed

sdk/ml/azure-ai-ml/tests/command_job/e2etests/test_command_job.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -488,6 +488,19 @@ def test_command_job_register_output(self, randstr: Callable[[str], str], client
488488
assert job.outputs.test3.name == "test3_output"
489489
assert job.outputs.test3.version == "3"
490490

491+
@pytest.mark.e2etest
492+
def test_ray_command_job(self, randstr: Callable[[], str], client: MLClient) -> None:
493+
job = client.jobs.create_or_update(
494+
load_job(
495+
source="./tests/test_configs/command_job/command_job_dist_ray.yml",
496+
params_override=[{"name": randstr("job_name")}],
497+
)
498+
)
499+
500+
job_status = wait_until_done(client=client, job=job)
501+
502+
assert job_status == JobStatus.COMPLETED
503+
491504

492505
def check_tid_in_url(client: MLClient, job: Job) -> None:
493506
# test that TID is placed in the URL only in live mode

0 commit comments

Comments
 (0)