Skip to content

Commit 7492f8c

Browse files
Update image name
1 parent 6ae4612 commit 7492f8c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/odh/mnist_ray_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ func mnistCustomRayImage(t *testing.T, numGpus int) {
6969
namespace := test.NewTestNamespace()
7070

7171
// Build and Push custom ray image
72-
image := "ray-torch"
72+
image := "ray-torch-cuda"
7373
buildAndPushRayImage(test, namespace.Name, image)
7474

7575
// Get custom ray image

0 commit comments

Comments
 (0)