Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 1 addition & 12 deletions ray-operator/test/support/environment.go
Original file line number Diff line number Diff line change
@@ -1,10 +1,7 @@
package support

import (
"fmt"
"os"
"runtime"
"strings"
)

const (
Expand All @@ -25,15 +22,7 @@ func GetRayVersion() string {
}

func GetRayImage() string {
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
return lookupEnvOrDefault(KuberayTestRayImage, RayImage)
}

func GetKubeRayUpgradeVersion() string {
Expand Down
Loading