diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index c8900212e5a1b..adb0879f20d47 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -12,7 +12,7 @@ ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH:-${PYTORCH_ROCM_ARCH}} RUN apt-get update -q -y && apt-get install -q -y \ sqlite3 libsqlite3-dev libfmt-dev libmsgpack-dev libsuitesparse-dev \ apt-transport-https ca-certificates wget curl -# Remove sccache +# Remove sccache RUN python3 -m pip install --upgrade pip RUN apt-get purge -y sccache; python3 -m pip uninstall -y sccache; rm -f "$(which sccache)" ARG COMMON_WORKDIR diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index 869fb28c3d85c..5172b39bb861b 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -29,4 +29,6 @@ matplotlib==3.10.3 # Multi-Modal Models Test (Extended) 3 blobfile==3.0.0 +schemathesis==3.39.15 # Required for openai schema test. +mteb[bm25s]>=1.38.11, <2 # required for mteb test