From c3c022165544a4bd386abcab07595a5a0a046fbf Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Fri, 26 Apr 2024 18:19:59 +0200 Subject: [PATCH] [push-ci-image] --- docker/consistency.dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile index 3fd141b05123c0..c4b3593771eeb1 100644 --- a/docker/consistency.dockerfile +++ b/docker/consistency.dockerfile @@ -6,7 +6,7 @@ ENV VIRTUAL_ENV=/usr/local RUN pip install uv RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools -RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras pytest pytest-xdist parameterized +RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras pytest pytest-xdist pytest-parametrized RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-cache-dir --upgrade GitPython RUN uv pip install --no-cache-dir "transformers[flax,quality]"