nits [push-ci-image]

huggingface · Apr 28, 2024 · aa62903 · aa62903
1 parent 28db44a
commit aa62903
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile
@@ -7,7 +7,6 @@ RUN pip install uv
 RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools GitPython
 RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu
-
 RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras
 RUN uv pip install --no-cache-dir "transformers[flax,quality,vision,testing]"
 

diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile
@@ -7,9 +7,9 @@ ENV VIRTUAL_ENV=/usr/local
 RUN pip --no-cache-dir install uv
 RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools
-RUN uv pip install --no-cache-dir "pytest<8.0.1" "fsspec>=2023.5.0,<2023.10.0" pytest-subtests pytest-xdist
+RUN uv pip install --no-cache-dir "fsspec>=2023.5.0,<2023.10.0"
 RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu
-RUN uv pip install --no-cache-dir "transformers[sklearn,sentencepiece,vision,timm]"
+RUN uv pip install --no-cache-dir "transformers[sklearn,sentencepiece,vision,timm,testing]"
 
 
 RUN pip uninstall -y transformers