From c435bc834aa30e595a245ea9a7001a5a8b5698ad Mon Sep 17 00:00:00 2001 From: artemvmin Date: Mon, 1 Apr 2024 13:57:22 -0700 Subject: [PATCH] Revert "Improve logic for polling Ray job status in jupyter notebook (#487)" This reverts commit acc09ef4fedffba357263c59183f1d58c51901fa. --- .../example_notebooks/rag-kaggle-ray-sql-latest.ipynb | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/applications/rag/example_notebooks/rag-kaggle-ray-sql-latest.ipynb b/applications/rag/example_notebooks/rag-kaggle-ray-sql-latest.ipynb index 3134e3287..89fd99c8a 100644 --- a/applications/rag/example_notebooks/rag-kaggle-ray-sql-latest.ipynb +++ b/applications/rag/example_notebooks/rag-kaggle-ray-sql-latest.ipynb @@ -283,16 +283,17 @@ " }\n", ")\n", "\n", - "# The Ray job typically takes 5m-10m to complete.\n", "print(\"Job submitted with ID:\", job_id)\n", + "prev_status = \"\"\n", "while True:\n", " status = client.get_job_status(job_id)\n", - " print(\"Job status:\", status)\n", - " print(\"Job info:\", client.get_job_info(job_id).message)\n", + " if status != prev_status:\n", + " print(\"Job status:\", status)\n", + " print(\"Job info:\", client.get_job_info(job_id).message)\n", + " prev_status = status\n", " if status.is_terminal():\n", " break\n", - " time.sleep(30)\n", - "\n", + " time.sleep(1)\n", "end_time = time.time()\n", "job_duration = end_time - start_time\n", "print(f\"Job completed in {job_duration} seconds.\")"