diff --git a/.github/ci-scripts/job_runner.py b/.github/ci-scripts/job_runner.py index 7229a59568..d5098475d2 100644 --- a/.github/ci-scripts/job_runner.py +++ b/.github/ci-scripts/job_runner.py @@ -2,8 +2,6 @@ import asyncio from pathlib import Path -from ray.job_submission import JobStatus, JobSubmissionClient - def parse_env_var_str(env_var_str: str) -> dict: iter = map( @@ -33,23 +31,25 @@ def submit_job( if args.enable_ray_tracing: env_vars["DAFT_ENABLE_RAY_TRACING"] = "1" - client = JobSubmissionClient(address="http://localhost:8265") - job_id = client.submit_job( - entrypoint=f"python {entrypoint_script} {entrypoint_args}", - runtime_env={ - "working_dir": working_dir, - "env_vars": env_vars, - }, - ) - - asyncio.run(wait_on_job(client.tail_job_logs(job_id), timeout_s=60 * 30)) - - status = client.get_job_status(job_id) - assert status.is_terminal(), "Job should have terminated" - if status != JobStatus.SUCCEEDED: - job_info = client.get_job_info(job_id) - raise RuntimeError(f"Job failed with {job_info.error_type} error: {job_info.message}") - print(f"Job completed with {status}") + print(env_vars) + + # client = JobSubmissionClient(address="http://localhost:8265") + # job_id = client.submit_job( + # entrypoint=f"python {entrypoint_script} {entrypoint_args}", + # runtime_env={ + # "working_dir": working_dir, + # "env_vars": env_vars, + # }, + # ) + + # asyncio.run(wait_on_job(client.tail_job_logs(job_id), timeout_s=60 * 30)) + + # status = client.get_job_status(job_id) + # assert status.is_terminal(), "Job should have terminated" + # if status != JobStatus.SUCCEEDED: + # job_info = client.get_job_info(job_id) + # raise RuntimeError(f"Job failed with {job_info.error_type} error: {job_info.message}") + # print(f"Job completed with {status}") if __name__ == "__main__":