Merge remote-tracking branch 'upstream/main'

dchourasia · dchourasia · commit 9a4979898c68 · 2024-09-11T01:05:55.000Z
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
