Update image name

opendatahub-io · Sep 10, 2024 · 7492f8c · 7492f8c
1 parent 6ae4612
commit 7492f8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/odh/mnist_ray_test.go b/tests/odh/mnist_ray_test.go
@@ -69,7 +69,7 @@ func mnistCustomRayImage(t *testing.T, numGpus int) {
 	namespace := test.NewTestNamespace()
 
 	// Build and Push custom ray image
-	image := "ray-torch"
+	image := "ray-torch-cuda"
 	buildAndPushRayImage(test, namespace.Name, image)
 
 	// Get custom ray image