From 01f8cede0b5f1c16330205b35f4391939e11cb3e Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Wed, 10 Jul 2024 23:05:36 +0200 Subject: [PATCH] Add koch to CI --- docker/lerobot-cpu/Dockerfile | 2 +- docker/lerobot-gpu/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/lerobot-cpu/Dockerfile b/docker/lerobot-cpu/Dockerfile index 7dd4c9f6..885a7752 100644 --- a/docker/lerobot-cpu/Dockerfile +++ b/docker/lerobot-cpu/Dockerfile @@ -21,7 +21,7 @@ RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc COPY . /lerobot WORKDIR /lerobot RUN pip install --upgrade --no-cache-dir pip -RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht]" \ +RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht, koch]" \ --extra-index-url https://download.pytorch.org/whl/cpu # Set EGL as the rendering backend for MuJoCo diff --git a/docker/lerobot-gpu/Dockerfile b/docker/lerobot-gpu/Dockerfile index 7058bf4d..86db693e 100644 --- a/docker/lerobot-gpu/Dockerfile +++ b/docker/lerobot-gpu/Dockerfile @@ -23,7 +23,7 @@ RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc COPY . /lerobot WORKDIR /lerobot RUN pip install --upgrade --no-cache-dir pip -RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht]" +RUN pip install --no-cache-dir ".[test, aloha, xarm, pusht, koch]" # Set EGL as the rendering backend for MuJoCo ENV MUJOCO_GL="egl"