remote ray client example

Co-authored-by: bkeane <[email protected]>
Co-authored-by: chipspeak <[email protected]>
Co-authored-by: Bobbins <[email protected]>

Author: 4 people

demo-notebooks/additional-demos/remote_ray_job_client.ipynb

@@ -0,0 +1,101 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Submit a job using an authorized Ray dashboard and the Job Submission Client remotely\n",
+    "In this notebook, we will demonstrate how to submit Ray jobs to an existing Raycluster, using the CodeFlare SDK.\n",
+    "\n",
+    "### Requirements\n",
+    "* Your administrator has created a RayCluster.\n",
+    "* They have provided you with its dashboard URL.\n",
+    "* You been provided an OpenShift authorization token."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Import pieces from codeflare-sdk\n",
+    "from codeflare_sdk import RayJobClient"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Setup Authentication Configuration \n",
+    "auth_token = \"XXXX\" # Replace with your actual token\n",
+    "header = {\n",
+    "    'Authorization': f'Bearer {auth_token}'\n",
+    "}"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Gather the dashboard URL (provided by the creator of the RayCluster)\n",
+    "ray_dashboard = \"XXXX\" # Replace with your Ray dashboard URL"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#Initialize the RayJobClient\n",
+    "client = RayJobClient(address=ray_dashboard, headers=header, verify=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Submit a job using the RayJobClient\n",
+    "entrypoint_command = \"python XXXX\" # Replace with your training script name\n",
+    "submission_id = client.submit_job(\n",
+    "    entrypoint=entrypoint_command,\n",
+    "    runtime_env={\"working_dir\": \"./\",\"pip\": \"requirements.txt\"},\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Get the job's status\n",
+    "client.get_job_status(submission_id)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Get the job's logs\n",
+    "client.get_job_logs(submission_id)"
+   ]
+  }
+ ],
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}