Incorporate SerialEnv and introduct multistep policy logic #90
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Test | |
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| types: [opened, synchronize, reopened, labeled] | |
| push: | |
| branches: | |
| - main | |
| jobs: | |
| test: | |
| if: | | |
| ${{ github.event_name == 'pull_request' && contains(github.event.pull_request.labels.*.name, 'CI') }} || | |
| ${{ github.event_name == 'push' }} | |
| runs-on: ubuntu-latest | |
| env: | |
| POETRY_VERSION: 1.8.1 | |
| DATA_DIR: tests/data | |
| TMPDIR: ~/tmp | |
| TEMP: ~/tmp | |
| TMP: ~/tmp | |
| PYOPENGL_PLATFORM: egl | |
| MUJOCO_GL: egl | |
| LEROBOT_TESTS_DEVICE: cpu | |
| steps: | |
| #---------------------------------------------- | |
| # check-out repo and set-up python | |
| #---------------------------------------------- | |
| - name: Check out repository | |
| uses: actions/checkout@v4 | |
| with: | |
| lfs: true | |
| - name: Set up python | |
| id: setup-python | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: '3.10' | |
| #---------------------------------------------- | |
| # install & configure poetry | |
| #---------------------------------------------- | |
| - name: Load cached Poetry installation | |
| id: restore-poetry-cache | |
| uses: actions/cache/restore@v3 | |
| with: | |
| path: ~/.local # the path depends on the OS | |
| key: poetry-${{ env.POETRY_VERSION }} # increment to reset cache | |
| - name: Install Poetry | |
| if: steps.restore-poetry-cache.outputs.cache-hit != 'true' | |
| uses: snok/install-poetry@v1 | |
| with: | |
| version: ${{ env.POETRY_VERSION }} | |
| virtualenvs-create: true | |
| installer-parallel: true | |
| - name: Save cached Poetry installation | |
| if: | | |
| steps.restore-poetry-cache.outputs.cache-hit != 'true' && | |
| github.ref_name == 'main' | |
| id: save-poetry-cache | |
| uses: actions/cache/save@v3 | |
| with: | |
| path: ~/.local # the path depends on the OS | |
| key: poetry-${{ env.POETRY_VERSION }} # increment to reset cache | |
| - name: Configure Poetry | |
| run: poetry config virtualenvs.in-project true | |
| #---------------------------------------------- | |
| # install dependencies | |
| #---------------------------------------------- | |
| - name: Load cached venv | |
| id: restore-dependencies-cache | |
| uses: actions/cache/restore@v3 | |
| with: | |
| path: .venv | |
| key: venv-${{ steps.setup-python.outputs.python-version }}-${{ env.POETRY_VERSION }}-${{ hashFiles('**/poetry.lock') }} | |
| - name: Info | |
| run: | | |
| sudo du -sh /tmp | |
| sudo df -h | |
| - name: Install dependencies | |
| if: steps.restore-dependencies-cache.outputs.cache-hit != 'true' | |
| run: | | |
| mkdir ~/tmp | |
| echo $TMPDIR | |
| echo $TEMP | |
| echo $TMP | |
| poetry install --no-interaction --no-root | |
| - name: Save cached venv | |
| if: | | |
| steps.restore-dependencies-cache.outputs.cache-hit != 'true' && | |
| github.ref_name == 'main' | |
| id: save-dependencies-cache | |
| uses: actions/cache/save@v3 | |
| with: | |
| path: .venv | |
| key: venv-${{ steps.setup-python.outputs.python-version }}-${{ env.POETRY_VERSION }}-${{ hashFiles('**/poetry.lock') }} | |
| - name: Install libegl1-mesa-dev (to use MUJOCO_GL=egl) | |
| run: sudo apt-get update && sudo apt-get install -y libegl1-mesa-dev | |
| #---------------------------------------------- | |
| # install project | |
| #---------------------------------------------- | |
| - name: Install project | |
| run: poetry install --no-interaction | |
| #---------------------------------------------- | |
| # run tests | |
| #---------------------------------------------- | |
| - name: Run tests | |
| run: | | |
| source .venv/bin/activate | |
| pytest tests | |
| - name: Test train pusht end-to-end | |
| run: | | |
| source .venv/bin/activate | |
| python lerobot/scripts/train.py \ | |
| hydra.job.name=pusht \ | |
| env=pusht \ | |
| wandb.enable=False \ | |
| offline_steps=2 \ | |
| online_steps=0 \ | |
| device=cpu \ | |
| save_model=true \ | |
| save_freq=1 \ | |
| hydra.run.dir=tests/outputs/ | |
| - name: Test eval pusht end-to-end | |
| run: | | |
| source .venv/bin/activate | |
| python lerobot/scripts/eval.py \ | |
| hydra.job.name=pusht \ | |
| env=pusht \ | |
| wandb.enable=False \ | |
| eval_episodes=1 \ | |
| env.episode_length=8 \ | |
| device=cpu \ | |
| policy.pretrained_model_path=tests/outputs/models/1.pt |