From bd754e200572408b7364eb6a05ce66c4241cac8d Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Fri, 19 Apr 2024 17:52:01 +0200
Subject: [PATCH] Test fixed builds

---
 .github/workflows/nightly-tests.yml | 78 ++++++++++++++---------------
 1 file changed, 39 insertions(+), 39 deletions(-)

diff --git a/.github/workflows/nightly-tests.yml b/.github/workflows/nightly-tests.yml
index 5b2e7edff..8167a65e9 100644
--- a/.github/workflows/nightly-tests.yml
+++ b/.github/workflows/nightly-tests.yml
@@ -3,9 +3,9 @@
 name: Self-hosted runner with slow tests (scheduled)
 
 on:
-  # pull_request:  # TODO(aliberts): for testing, delete before merge
-  #   branches:
-  #     - main
+  pull_request:  # TODO(aliberts): for testing, delete before merge
+    branches:
+      - main
   workflow_dispatch:
   schedule:
     - cron: "0 2 * * *"
@@ -46,10 +46,10 @@ jobs:
           ls -glh /opt/venv/
           ls -glh /opt/venv/lib
           ls -glh /opt/venv/lib/python3.10/site-packages
-      - name: Pip install
-        run: |
-          source activate peft
-          pip install ".[test, aloha, xarm, pusht]"
+      # - name: Pip install
+      #   run: |
+      #     source activate peft
+      #     pip install ".[test, aloha, xarm, pusht]"
       - name: Test
         env:
           DATA_DIR: tests/data
@@ -57,38 +57,38 @@ jobs:
         run: |
           pytest -v --cov=./lerobot --cov-report=xml tests
 
-  # run_all_tests_single_gpu:
-  #   strategy:
-  #     fail-fast: false
-  #   runs-on: [single-gpu, nvidia-gpu, t4, ci]
-  #   env:
-  #     CUDA_VISIBLE_DEVICES: "0"
-  #     TEST_TYPE: "single_gpu"
-  #   container:
-  #     image: huggingface/lerobot-gpu:latest
-  #     options: --gpus all --shm-size "16gb" -e NVIDIA_DISABLE_REQUIRE=true
-  #     credentials:
-  #       username: ${{ secrets.DOCKERHUB_USERNAME }}
-  #       password: ${{ secrets.DOCKERHUB_PASSWORD }}
-  #   defaults:
-  #     run:
-  #       shell: bash
-  #   steps:
-  #   #   - uses: actions/checkout@v4
-  #     - name: Check
-  #       run: |
-  #         pwd
-  #         df -h
-  #         ls -glh
-  #         du -sh
-  #         python --version
-  #         which python
-  #     - name: Test
-  #       env:
-  #         DATA_DIR: tests/data
-  #         MUJOCO_GL: egl
-  #       run: |
-  #         pytest -v --cov=./lerobot --cov-report=xml tests
+  run_all_tests_single_gpu:
+    strategy:
+      fail-fast: false
+    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    env:
+      CUDA_VISIBLE_DEVICES: "0"
+      TEST_TYPE: "single_gpu"
+    container:
+      image: huggingface/lerobot-gpu:latest
+      options: --gpus all --shm-size "16gb" -e NVIDIA_DISABLE_REQUIRE=true
+      credentials:
+        username: ${{ secrets.DOCKERHUB_USERNAME }}
+        password: ${{ secrets.DOCKERHUB_PASSWORD }}
+    defaults:
+      run:
+        shell: bash
+    steps:
+    #   - uses: actions/checkout@v4
+      - name: Check
+        run: |
+          pwd
+          df -h
+          ls -glh
+          du -sh
+          python --version
+          which python
+      - name: Test
+        env:
+          DATA_DIR: tests/data
+          MUJOCO_GL: egl
+        run: |
+          pytest -v --cov=./lerobot --cov-report=xml tests
 
     #   - name: Generate Report
     #     if: always()