e2e tests: Fix finishing condition in ODH e2e test

project-codeflare · Mar 11, 2024 · 7419586 · 7419586
1 parent bb24607
commit 7419586
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/odh/resources/mnist_mcad_mini.ipynb b/test/odh/resources/mnist_mcad_mini.ipynb
@@ -50,7 +50,7 @@
     "while not finished:\n",
     "    sleep(1)\n",
     "    try:\n",
-    "        finished = (\"Epoch 4: 100%\" in job.logs())\n",
+    "        finished = (\"Epoch 2: 100%\" in job.logs())\n",
     "    except:\n",
     "        finished = False"
    ]