From c26f946f26b749177ac4b001c730f12136faeacb Mon Sep 17 00:00:00 2001
From: "Alexei V. Ivanov" <alexei.ivanov@amd.com>
Date: Wed, 22 Jan 2025 13:56:47 -0600
Subject: [PATCH] Fixing the LoRA CI test.

Signed-off-by: Alexei V. Ivanov <alexei.ivanov@amd.com>
---
 Dockerfile.rocm | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 7213a15a2e00..14c522afd7f9 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -72,7 +72,8 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm /vllm-workspace
 RUN cd /vllm-workspace \
     && rm -rf vllm \
     && python3 -m pip install -e tests/vllm_test_utils \
-    && python3 -m pip install lm-eval[api]==0.4.4
+    && python3 -m pip install lm-eval[api]==0.4.4 \
+    && python3 -m pip install pytest-shard
 
 # -----------------------
 # Final vLLM image