From 8194b73645a9188deeab3b2a3a226fd63c87bc56 Mon Sep 17 00:00:00 2001
From: Sivanantham Chinnaiyan <sivanantham.chinnaiyan@ideas2it.com>
Date: Wed, 26 Jun 2024 21:32:14 +0530
Subject: [PATCH] debug

Signed-off-by: Sivanantham Chinnaiyan <sivanantham.chinnaiyan@ideas2it.com>
---
 test/e2e/predictor/test_huggingface.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/test/e2e/predictor/test_huggingface.py b/test/e2e/predictor/test_huggingface.py
index 87fa0191de9..090393541ef 100644
--- a/test/e2e/predictor/test_huggingface.py
+++ b/test/e2e/predictor/test_huggingface.py
@@ -16,7 +16,7 @@
 
 import pytest
 from kubernetes import client
-from kubernetes.client import V1ResourceRequirements
+from kubernetes.client import V1ResourceRequirements, V1EnvVar
 
 from kserve import (
     V1beta1PredictorSpec,
@@ -302,6 +302,13 @@ def test_vllm_openai_chat_completions():
                 "--device",
                 "cpu",
             ],
+            env=[
+                V1EnvVar(
+                    "VLLM_LOGGING_LEVEL",
+                    "DEBUG",
+                ),
+                V1EnvVar("VLLM_TRACE_FUNCTION", "1"),
+            ],  # Add environment variables
             resources=V1ResourceRequirements(
                 requests={"cpu": "1", "memory": "2Gi"},
                 limits={"cpu": "1", "memory": "4Gi"},