From 43e079f73e21602374e3707e3839dfd290ccaf19 Mon Sep 17 00:00:00 2001
From: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Date: Sun, 2 Feb 2025 13:59:33 +0100
Subject: [PATCH] Fix nightly tests docker images (#675)

---
 docker/lerobot-cpu/Dockerfile | 5 +++--
 docker/lerobot-gpu/Dockerfile | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/docker/lerobot-cpu/Dockerfile b/docker/lerobot-cpu/Dockerfile
index c725c411..06673092 100644
--- a/docker/lerobot-cpu/Dockerfile
+++ b/docker/lerobot-cpu/Dockerfile
@@ -7,7 +7,7 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 # Install apt dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential cmake git \
+    build-essential cmake git git-lfs \
     libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
     speech-dispatcher libgeos-dev \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
@@ -19,7 +19,8 @@ ENV PATH="/opt/venv/bin:$PATH"
 RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
 
 # Install LeRobot
-RUN GIT_LFS_SKIP_SMUDGE=1 git clone https://github.com/huggingface/lerobot.git /lerobot
+RUN git lfs install
+RUN git clone https://github.com/huggingface/lerobot.git /lerobot
 WORKDIR /lerobot
 RUN pip install --upgrade --no-cache-dir pip
 RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht, dynamixel]" \
diff --git a/docker/lerobot-gpu/Dockerfile b/docker/lerobot-gpu/Dockerfile
index b362fe8a..65ca4377 100644
--- a/docker/lerobot-gpu/Dockerfile
+++ b/docker/lerobot-gpu/Dockerfile
@@ -7,7 +7,7 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 # Install apt dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential cmake git \
+    build-essential cmake git git-lfs \
     libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
     speech-dispatcher libgeos-dev \
     python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv \
@@ -21,7 +21,8 @@ ENV PATH="/opt/venv/bin:$PATH"
 RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
 
 # Install LeRobot
-RUN GIT_LFS_SKIP_SMUDGE=1 git clone https://github.com/huggingface/lerobot.git /lerobot
+RUN git lfs install
+RUN git clone https://github.com/huggingface/lerobot.git /lerobot
 WORKDIR /lerobot
 RUN pip install --upgrade --no-cache-dir pip
 RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht, dynamixel]"