From 980bc0fe9cab9b78cd3cc632ea24347600629164 Mon Sep 17 00:00:00 2001
From: Milind Waykole <mwaykole@mwaykole-thinkpadp1gen4i.bengluru.csb>
Date: Thu, 30 Jan 2025 15:25:08 +0530
Subject: [PATCH] addressed comments

Signed-off-by: Milind Waykole <mwaykole@mwaykole-thinkpadp1gen4i.bengluru.csb>
---
 tests/model_serving/model_server/authentication/conftest.py | 3 ++-
 tests/model_serving/model_server/utils.py                   | 5 +++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/model_serving/model_server/authentication/conftest.py b/tests/model_serving/model_server/authentication/conftest.py
index 220cbf81..0e3aa5d9 100644
--- a/tests/model_serving/model_server/authentication/conftest.py
+++ b/tests/model_serving/model_server/authentication/conftest.py
@@ -26,6 +26,7 @@
 )
 from utilities.serving_runtime import ServingRuntimeFromTemplate
 from utilities.constants import Annotations
+from utilities.constants import Labels
 
 
 # GRPC model serving
@@ -181,7 +182,7 @@ def patched_remove_raw_authentication_isvc(
         patches={
             http_s3_caikit_raw_inference_service: {
                 "metadata": {
-                    "annotations": {Annotations.KserveAuth.SECURITY: "false"},
+                    "labels": {Labels.KserveAuth.SECURITY: "false"},
                 }
             }
         }
diff --git a/tests/model_serving/model_server/utils.py b/tests/model_serving/model_server/utils.py
index 252c5bd9..719283f0 100644
--- a/tests/model_serving/model_server/utils.py
+++ b/tests/model_serving/model_server/utils.py
@@ -12,6 +12,7 @@
 from utilities.constants import (
     Annotations,
     KServeDeploymentType,
+    Labels,
 )
 from utilities.exceptions import (
     FailedPodsError,
@@ -129,9 +130,9 @@ def create_isvc(
     if enable_auth:
         # model mesh auth is set in servingruntime
         if deployment_mode == KServeDeploymentType.SERVERLESS:
-            annotations["security.opendatahub.io/enable-auth"] = "true"
+            annotations[Annotations.KserveAuth.SECURITY] = "true"
         elif deployment_mode == KServeDeploymentType.RAW_DEPLOYMENT:
-            labels["security.opendatahub.io/enable-auth"] = "true"
+            labels[Labels.KserveAuth.SECURITY] = "true"
 
     # default to True if deployment_mode is Serverless (default behavior of Serverless) if was not provided by the user
     # model mesh external route is set in servingruntime