Remove default fms-hf-tuning image

Author: sutaakar

tests/kfto/core/environment.go

@@ -18,6 +18,8 @@ package core
 
 import (
 	"os"
+
+	. "github.com/project-codeflare/codeflare-common/support"
 )
 
 const (
@@ -27,8 +29,12 @@ const (
 	bloomModelImageEnvVar = "BLOOM_MODEL_IMAGE"
 )
 
-func GetFmsHfTuningImage() string {
-	return lookupEnvOrDefault(fmsHfTuningImageEnvVar, "quay.io/modh/fms-hf-tuning:bd8bf628cd739c7a201a976bc3c1096785353f1a")
+func GetFmsHfTuningImage(t Test) string {
+	image, ok := os.LookupEnv(fmsHfTuningImageEnvVar)
+	if !ok {
+		t.T().Fatalf("Expected environment variable %s not found, please use this environment variable to specify fms-hf-tuning image to be tested.", fmsHfTuningImageEnvVar)
+	}
+	return image
 }
 
 func GetBloomModelImage() string {

tests/kfto/core/kfto_kueue_sft_test.go

@@ -200,7 +200,7 @@ func createPyTorchJob(test Test, namespace, localQueueName string, config corev1
 							Containers: []corev1.Container{
 								{
 									Name:            "pytorch",
-									Image:           GetFmsHfTuningImage(),
+									Image:           GetFmsHfTuningImage(test),
 									ImagePullPolicy: corev1.PullIfNotPresent,
 									Env: []corev1.EnvVar{
 										{

tests/kfto/upgrade/kfto_kueue_sft_upgrade_test.go

@@ -164,7 +164,7 @@ func createPyTorchJob(test Test, namespace, localQueueName string, config corev1
 							Containers: []corev1.Container{
 								{
 									Name:            "pytorch",
-									Image:           kftocore.GetFmsHfTuningImage(),
+									Image:           kftocore.GetFmsHfTuningImage(test),
 									ImagePullPolicy: corev1.PullIfNotPresent,
 									Env: []corev1.EnvVar{
 										{