[Feature] Support ARM image for test (#2699)

ray-operator/test/support/environment.go

@@ -1,7 +1,10 @@
 package support
 
 import (
+	"fmt"
 	"os"
+	"runtime"
+	"strings"
 )
 
 const (
@@ -20,7 +23,15 @@ func GetRayVersion() string {
 }
 
 func GetRayImage() string {
-	return lookupEnvOrDefault(KuberayTestRayImage, RayImage)
+	rayImage := lookupEnvOrDefault(KuberayTestRayImage, RayImage)
+	// detect if we are running on arm64 machine, most likely apple silicon
+	// the os name is not checked as it also possible that it might be linux
+	// also check if the image does not have the `-aarch64` suffix
+	if runtime.GOARCH == "arm64" && !strings.HasSuffix(rayImage, "-aarch64") {
+		rayImage = rayImage + "-aarch64"
+		fmt.Printf("Modified Ray Image to: %s for ARM chips\n", rayImage)
+	}
+	return rayImage
 }
 
 func lookupEnvOrDefault(key, value string) string {