From 678532c16935712aa61379938916b31e1dcee8b8 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Tue, 21 May 2024 14:05:23 +0200 Subject: [PATCH] [build-ci-image] --- docker/consistency.dockerfile | 4 ++-- docker/quality.dockerfile | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile index fa94259e2dedae..48844fcc4d1861 100644 --- a/docker/consistency.dockerfile +++ b/docker/consistency.dockerfile @@ -3,8 +3,8 @@ ENV PYTHONDONTWRITEBYTECODE=1 USER root RUN apt-get update && apt-get install -y time git pkg-config make git-lfs ENV VIRTUAL_ENV=/usr/local -RUN pip install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools GitPython -RUN uv pip install --no-cache-dir --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu +RUN pip install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools GitPython ruff==0.4.4 +uUN uv pip install --no-cache-dir --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]" RUN git lfs install diff --git a/docker/quality.dockerfile b/docker/quality.dockerfile index 6e31b02eb7dcb1..392a254d515754 100644 --- a/docker/quality.dockerfile +++ b/docker/quality.dockerfile @@ -4,5 +4,5 @@ USER root RUN apt-get update && apt-get install -y time git ENV VIRTUAL_ENV=/usr/local RUN pip install uv && uv venv -RUN uv pip install --no-cache-dir -U pip setuptools GitPython "transformers[ruff]" urllib3 +RUN uv pip install --no-cache-dir -U pip setuptools GitPython "ruff==0.4.4" transformers urllib3 RUN apt-get install -y jq curl && apt-get clean && rm -rf /var/lib/apt/lists/* \ No newline at end of file