Skip to content

Commit c9a04d8

Browse files
authored
Fix nightly (huggingface#775)
1 parent 4fd9fdd commit c9a04d8

File tree

51 files changed

+144
-171
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

51 files changed

+144
-171
lines changed

.github/workflows/test-docker-build.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ jobs:
4343
needs: get_changed_files
4444
runs-on:
4545
group: aws-general-8-plus
46-
if: ${{ needs.get_changed_files.outputs.matrix }} != ''
46+
if: needs.get_changed_files.outputs.matrix != ''
4747
strategy:
4848
fail-fast: false
4949
matrix:

docker/lerobot-cpu/Dockerfile

Lines changed: 16 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,33 +1,29 @@
11
# Configure image
22
ARG PYTHON_VERSION=3.10
3-
43
FROM python:${PYTHON_VERSION}-slim
4+
5+
# Configure environment variables
56
ARG PYTHON_VERSION
6-
ARG DEBIAN_FRONTEND=noninteractive
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
ENV MUJOCO_GL="egl"
9+
ENV PATH="/opt/venv/bin:$PATH"
710

8-
# Install apt dependencies
11+
# Install dependencies and set up Python in a single layer
912
RUN apt-get update && apt-get install -y --no-install-recommends \
10-
build-essential cmake git git-lfs \
13+
build-essential cmake git \
1114
libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
1215
speech-dispatcher libgeos-dev \
13-
&& apt-get clean && rm -rf /var/lib/apt/lists/*
16+
&& ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
17+
&& python -m venv /opt/venv \
18+
&& apt-get clean && rm -rf /var/lib/apt/lists/* \
19+
&& echo "source /opt/venv/bin/activate" >> /root/.bashrc
1420

15-
# Create virtual environment
16-
RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
17-
RUN python -m venv /opt/venv
18-
ENV PATH="/opt/venv/bin:$PATH"
19-
RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
20-
21-
# Install LeRobot
22-
RUN git lfs install
23-
RUN git clone https://github.com/huggingface/lerobot.git /lerobot
21+
# Clone repository and install LeRobot in a single layer
22+
COPY . /lerobot
2423
WORKDIR /lerobot
25-
RUN pip install --upgrade --no-cache-dir pip
26-
RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht, dynamixel]" \
27-
--extra-index-url https://download.pytorch.org/whl/cpu
28-
29-
# Set EGL as the rendering backend for MuJoCo
30-
ENV MUJOCO_GL="egl"
24+
RUN /opt/venv/bin/pip install --upgrade --no-cache-dir pip \
25+
&& /opt/venv/bin/pip install --no-cache-dir ".[test, aloha, xarm, pusht, dynamixel]" \
26+
--extra-index-url https://download.pytorch.org/whl/cpu
3127

3228
# Execute in bash shell rather than python
3329
CMD ["/bin/bash"]

docker/lerobot-gpu/Dockerfile

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ ENV PATH="/opt/venv/bin:$PATH"
88

99
# Install dependencies and set up Python in a single layer
1010
RUN apt-get update && apt-get install -y --no-install-recommends \
11-
build-essential cmake git git-lfs \
11+
build-essential cmake git \
1212
libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
1313
speech-dispatcher libgeos-dev \
1414
python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv \
@@ -18,8 +18,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
1818
&& echo "source /opt/venv/bin/activate" >> /root/.bashrc
1919

2020
# Clone repository and install LeRobot in a single layer
21+
COPY . /lerobot
2122
WORKDIR /lerobot
22-
RUN git lfs install \
23-
&& git clone https://github.com/huggingface/lerobot.git . \
24-
&& /opt/venv/bin/pip install --upgrade --no-cache-dir pip \
23+
RUN /opt/venv/bin/pip install --upgrade --no-cache-dir pip \
2524
&& /opt/venv/bin/pip install --no-cache-dir ".[test, aloha, xarm, pusht, dynamixel]"

tests/data/save_policy_to_safetensors/aloha_act/actions.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

tests/data/save_policy_to_safetensors/aloha_act/grad_stats.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

tests/data/save_policy_to_safetensors/aloha_act/output_dict.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

tests/data/save_policy_to_safetensors/aloha_act/param_stats.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

tests/data/save_policy_to_safetensors/aloha_act_1000_steps/actions.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

tests/data/save_policy_to_safetensors/aloha_act_1000_steps/grad_stats.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

tests/data/save_policy_to_safetensors/aloha_act_1000_steps/output_dict.safetensors

Lines changed: 0 additions & 3 deletions
This file was deleted.

0 commit comments

Comments
 (0)