From 2f6af1abff5a1f77d2fa538e7ff6c56c4c6e3737 Mon Sep 17 00:00:00 2001
From: Rui Vieira <rcardoso@redhat.com>
Date: Thu, 28 Nov 2024 14:27:01 +0000
Subject: [PATCH] feat(lmeval): Disable online evaluation (#373)

---
 controllers/lmes/lmevaljob_controller.go      |  4 +++
 controllers/lmes/lmevaljob_controller_test.go | 36 +++++++++++++++++++
 2 files changed, 40 insertions(+)

diff --git a/controllers/lmes/lmevaljob_controller.go b/controllers/lmes/lmevaljob_controller.go
index ce48ddc9..88837454 100644
--- a/controllers/lmes/lmevaljob_controller.go
+++ b/controllers/lmes/lmevaljob_controller.go
@@ -740,6 +740,10 @@ func CreatePod(svcOpts *serviceOptions, job *lmesv1alpha1.LMEvalJob, log logr.Lo
 			Name:  "TRANSFORMERS_OFFLINE",
 			Value: "1",
 		},
+		{
+			Name:  "HF_EVALUATE_OFFLINE",
+			Value: "1",
+		},
 	}
 	envVars = append(envVars, offlineHuggingFaceEnvVars...)
 
diff --git a/controllers/lmes/lmevaljob_controller_test.go b/controllers/lmes/lmevaljob_controller_test.go
index 3620471a..908231ea 100644
--- a/controllers/lmes/lmevaljob_controller_test.go
+++ b/controllers/lmes/lmevaljob_controller_test.go
@@ -137,6 +137,10 @@ func Test_SimplePod(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 				},
 			},
@@ -341,6 +345,10 @@ func Test_WithCustomPod(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 				},
 				{
@@ -527,6 +535,10 @@ func Test_EnvSecretsPod(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 					Command:         generateCmd(svcOpts, job),
 					Args:            generateArgs(svcOpts, job, log),
@@ -677,6 +689,10 @@ func Test_FileSecretsPod(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 					VolumeMounts: []corev1.VolumeMount{
 						{
@@ -1126,6 +1142,10 @@ func Test_ManagedPVC(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 
 					VolumeMounts: []corev1.VolumeMount{
@@ -1265,6 +1285,10 @@ func Test_ExistingPVC(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 					VolumeMounts: []corev1.VolumeMount{
 						{
@@ -1421,6 +1445,10 @@ func Test_PVCPreference(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 					VolumeMounts: []corev1.VolumeMount{
 						{
@@ -1608,6 +1636,10 @@ func Test_OfflineMode(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 					VolumeMounts: []corev1.VolumeMount{
 						{
@@ -1772,6 +1804,10 @@ func Test_OfflineModeWithOutput(t *testing.T) {
 							Name:  "TRANSFORMERS_OFFLINE",
 							Value: "1",
 						},
+						{
+							Name:  "HF_EVALUATE_OFFLINE",
+							Value: "1",
+						},
 					},
 					VolumeMounts: []corev1.VolumeMount{
 						{