From a0bf644e4525f97215831844f1762c5a6894765e Mon Sep 17 00:00:00 2001
From: Karel Suta <ksuta@redhat.com>
Date: Mon, 19 Feb 2024 15:33:12 +0100
Subject: [PATCH] e2e tests: Use test image from
 quay.io/project-codeflare/demo-images

---
 test/odh/resources/mnist_mcad_mini.ipynb | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/odh/resources/mnist_mcad_mini.ipynb b/test/odh/resources/mnist_mcad_mini.ipynb
index 0b53324ab..b32a95ab9 100644
--- a/test/odh/resources/mnist_mcad_mini.ipynb
+++ b/test/odh/resources/mnist_mcad_mini.ipynb
@@ -19,7 +19,9 @@
    "execution_count": null,
    "id": "47ca5c15",
    "metadata": {
-    "tags": ["parameters"]
+    "tags": [
+     "parameters"
+    ]
    },
    "outputs": [],
    "source": [
@@ -34,7 +36,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "job = DDPJobDefinition(name=\"mnistjob\", script=\"mnist.py\", scheduler_args={\"namespace\": namespace}, j=\"1x1\", gpu=0, cpu=1, memMB=2000, image=\"quay.io/project-codeflare/mnist-job-test:v0.0.1\").submit()"
+    "job = DDPJobDefinition(name=\"mnistjob\", script=\"/test/mnist.py\", scheduler_args={\"namespace\": namespace}, j=\"1x1\", gpu=0, cpu=1, memMB=2000, image=\"quay.io/project-codeflare/demo-images:pytorch-mnist-v0.0.1\").submit()"
    ]
   },
   {