Fix uses path

Signed-off-by: sailesh duddupudi <[email protected]>
kubeflow · Sep 29, 2024 · 34be848 · 34be848
1 parent 222bf0e
commit 34be848
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/.github/workflows/test-example-notebooks.yaml b/.github/workflows/test-example-notebooks.yaml
@@ -17,7 +17,7 @@ jobs:
         python-version: ["3.9", "3.10", "3.11"]
     steps:
       - name: Test Notebook
-        uses: .github/workflows/template-notebook-test
+        uses: ./.github/workflows/template-notebook-test
         with:
           kubernetes-version: ${{ matrix.kubernetes-version }}
           python-version: ${{ matrix.python-version }}

diff --git a/examples/pytorch/image-classification/create-pytorchjob.ipynb b/examples/pytorch/image-classification/create-pytorchjob.ipynb
@@ -129,10 +129,14 @@
     "        metadata=V1ObjectMeta(\n",
     "            name=name,\n",
     "            namespace=namespace,\n",
-    "            annotations={\"sidecar.istio.io/inject\": \"false\"},\n",
+    "            annotations={\n",
+    "                \"sidecar.istio.io/inject\": \"false\"\n",
+    "            }\n",
     "        ),\n",
-    "        spec=V1PodSpec(containers=[container]),\n",
-    "    ),\n",
+    "        spec=V1PodSpec(\n",
+    "            containers=[container]\n",
+    "        )\n",
+    "    )\n",
     ")\n",
     "\n",
     "pytorchjob = KubeflowOrgV1PyTorchJob(\n",
@@ -141,7 +145,10 @@
     "    metadata=V1ObjectMeta(name=name, namespace=namespace),\n",
     "    spec=KubeflowOrgV1PyTorchJobSpec(\n",
     "        run_policy=KubeflowOrgV1RunPolicy(clean_pod_policy=\"None\"),\n",
-    "        pytorch_replica_specs={\"Master\": replica_spec, \"Worker\": replica_spec},\n",
+    "        pytorch_replica_specs={\n",
+    "            \"Master\": replica_spec,\n",
+    "            \"Worker\": replica_spec\n",
+    "        },\n",
     "    ),\n",
     ")"
    ]
@@ -308,9 +315,7 @@
    "source": [
     "pytorchjob = training_client.wait_for_job_conditions(name=name)\n",
     "\n",
-    "print(\n",
-    "    f\"Succeeded number of replicas: {pytorchjob.status.replica_statuses['Master'].succeeded}\"\n",
-    ")"
+    "print(f\"Succeeded number of replicas: {pytorchjob.status.replica_statuses['Master'].succeeded}\")"
    ]
   },
   {