Update outdated interactive

project-codeflare · Aug 11, 2023 · 329ef33 · 329ef33
1 parent 5ccca0d
commit 329ef33
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/demo-notebooks/additional-interactive/hf_interactive.ipynb b/demo-notebooks/additional-interactive/hf_interactive.ipynb
@@ -53,11 +53,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Create authentication object for oc user permissions and login\n",
+    "# Create authentication object for user permissions\n",
+    "# IF unused, SDK will automatically check for default kubeconfig, then in-cluster config\n",
+    "# KubeConfigFileAuthentication can also be used to specify kubeconfig path manually\n",
     "auth = TokenAuthentication(\n",
     "    token = \"XXXX\",\n",
     "    server = \"XXXX\",\n",
-    "    skip_tls = True\n",
+    "    skip_tls = False\n",
     ")\n",
     "auth.login()"
    ]
@@ -88,13 +90,12 @@
     "# Create our cluster and submit appwrapper\n",
     "cluster = Cluster(ClusterConfiguration(name='hfgputest', \n",
     "                                       namespace=\"default\",\n",
-    "                                       min_worker=1, \n",
-    "                                       max_worker=1, \n",
+    "                                       num_workers=1,\n",
     "                                       min_cpus=8, \n",
     "                                       max_cpus=8, \n",
     "                                       min_memory=16, \n",
     "                                       max_memory=16, \n",
-    "                                       gpu=4,\n",
+    "                                       num_gpus=4,\n",
     "                                       image=\"quay.io/project-codeflare/ray:2.5.0-py38-cu116\",\n",
     "                                       instascale=True, machine_types=[\"m5.xlarge\", \"p3.8xlarge\"]))"
    ]
@@ -319,7 +320,7 @@
     "#install additional libraries that will be required for this training\n",
     "runtime_env = {\"pip\": [\"transformers\", \"datasets\", \"evaluate\", \"pyarrow<7.0.0\", \"accelerate\"]}\n",
     "\n",
-    "ray.init(address=f'{ray_cluster_uri}', runtime_env=runtime_env)\n",
+    "ray.init(address=ray_cluster_uri, runtime_env=runtime_env)\n",
     "\n",
     "print(\"Ray cluster is up and running: \", ray.is_initialized())"
    ]