From ccaa259de66e6ad0ef68b98ec0fa0588fd57373e Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Sun, 28 Apr 2024 19:03:55 +0200 Subject: [PATCH] nits to run tests, no push-ci --- docker/torch-tf-light.dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile index 9c28ef6dd600f5..f8bc5fb6a9b941 100644 --- a/docker/torch-tf-light.dockerfile +++ b/docker/torch-tf-light.dockerfile @@ -11,7 +11,7 @@ RUN uv pip install --no-cache-dir -U pip setuptools RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-cache-dir pypi-kenlm protobuf==3.20.3 accelerate tensorflow_probability pytest pytest-xdist parameterized -RUN uv pip install --no-cache-dir "transformers[sklearn,tf-cpu,sentencepiece,vision]" +RUN uv pip install --no-cache-dir "transformers[sklearn,tf-cpu,sentencepiece,vision, testing]" RUN pip uninstall -y transformers