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,90 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Submit a job using an authorized Ray dashboard and the Job Submission Client remotely"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "vscode": {
+     "languageId": "plaintext"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "# Import pieces from codeflare-sdk\n",
+    "from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication, RayJobClient\n",
+    "\n",
+    "# Gather the dashboard URL\n",
+    "ray_dashboard = \"<Ray-Dashboard-Route>\"\n",
+    "\n",
+    "# Setup Authentication Configuration \n",
+    "\n",
+    "header = {\n",
+    "    'Authorization': f'Bearer {<auth-token>}'\n",
+    "}\n",
+    "\n",
+    "#Initialize the RayJobClient\n",
+    "client = RayJobClient(address=ray_dashboard, headers=header, verify=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "vscode": {
+     "languageId": "plaintext"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "# Submit an a job using the RayJobClient\n",
+    "entrypoint_command = \"python <training-script>\"\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": {
+    "vscode": {
+     "languageId": "plaintext"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "# Get the job's status\n",
+    "client.get_job_status(submission_id)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "vscode": {
+     "languageId": "plaintext"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "# Get the job's logs\n",
+    "client.get_job_logs(submission_id)"
+   ]
+  }
+ ],
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}