Add PPO + Transformer-XL #1602
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: tests | |
on: | |
pull_request: | |
paths-ignore: | |
- '**/README.md' | |
- 'docs/**/*' | |
- 'cloud/**/*' | |
jobs: | |
test-core-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04, windows-latest] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# classic control tests | |
- name: Install core dependencies | |
run: poetry install -E pytest | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Run core tests | |
run: poetry run pytest tests/test_classic_control.py | |
- name: Install jax | |
if: runner.os == 'Linux' || runner.os == 'macOS' | |
run: poetry install -E "pytest jax" | |
- name: Run gymnasium migration dependencies | |
run: poetry run pip install "stable_baselines3==2.0.0a1" | |
- name: Run gymnasium tests | |
run: poetry run pytest tests/test_classic_control_gymnasium.py | |
- name: Run core tests with jax | |
if: runner.os == 'Linux' || runner.os == 'macOS' | |
run: poetry run pytest tests/test_classic_control_jax_gymnasium.py | |
- name: Run gae tests with jax | |
if: runner.os == 'Linux' || runner.os == 'macOS' | |
run: poetry run pytest tests/test_jax_compute_gae.py | |
- name: Install tuner dependencies | |
run: poetry install -E "pytest optuna" | |
- name: Run tuner tests | |
run: poetry run pytest tests/test_tuner.py | |
test-atari-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04, windows-latest] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# atari tests | |
- name: Install atari dependencies | |
run: poetry install -E "pytest atari" | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Run atari tests | |
run: poetry run pytest tests/test_atari.py | |
- name: Install jax | |
if: runner.os == 'Linux' || runner.os == 'macOS' | |
run: poetry install -E "pytest atari jax" | |
- name: Run gymnasium migration dependencies | |
run: poetry run pip install "stable_baselines3==2.0.0a1" "gymnasium[atari,accept-rom-license]==0.28.1" "ale-py==0.8.1" | |
- name: Run gymnasium tests | |
run: poetry run pytest tests/test_atari_gymnasium.py | |
- name: Run gymnasium tests with jax | |
if: runner.os == 'Linux' || runner.os == 'macOS' | |
run: poetry run pytest tests/test_atari_jax_gymnasium.py | |
test-procgen-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04, windows-latest] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# procgen tests | |
- name: Install core dependencies | |
run: poetry install -E "pytest procgen" | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Run procgen tests | |
run: poetry run pytest tests/test_procgen.py | |
test-mujoco-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# mujoco tests | |
- name: Install dependencies | |
run: poetry install -E "pytest mujoco dm_control jax" | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Run gymnasium migration dependencies | |
run: poetry run pip install "stable_baselines3==2.0.0a1" | |
- name: install mujoco dependencies | |
run: | | |
sudo apt-get update && sudo apt-get -y install libgl1-mesa-glx libosmesa6 libglfw3 | |
- name: Run mujoco tests | |
continue-on-error: true # MUJOCO_GL=osmesa results in `free(): invalid pointer` | |
run: poetry run pytest tests/test_mujoco.py | |
# test-mujoco-envs-mac: | |
# strategy: | |
# fail-fast: false | |
# matrix: | |
# python-version: [3.8] | |
# poetry-version: [1.3.1] | |
# os: [macos-latest] | |
# runs-on: ${{ matrix.os }} | |
# steps: | |
# - uses: actions/checkout@v2 | |
# - uses: actions/setup-python@v2 | |
# with: | |
# python-version: ${{ matrix.python-version }} | |
# - name: Run image | |
# uses: abatilo/actions-poetry@v2.0.0 | |
# with: | |
# poetry-version: ${{ matrix.poetry-version }} | |
# # mujoco tests | |
# - name: Install dependencies | |
# run: poetry install -E "pytest mujoco dm_control jax" | |
# - name: Downgrade setuptools | |
# run: poetry run pip install setuptools==59.5.0 | |
# - name: Run gymnasium migration dependencies | |
# run: poetry run pip install "stable_baselines3==2.0.0a1" | |
# - name: Run mujoco tests | |
# run: poetry run pytest tests/test_mujoco.py | |
test-mujoco_py-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
test-envpool-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# envpool tests | |
- name: Install envpool dependencies | |
run: poetry install -E "pytest envpool jax ppo_atari_envpool_xla_jax_scan" | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Run envpool tests | |
run: poetry run pytest tests/test_envpool.py | |
test-atari-multigpu-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# atari multigpu tests | |
- name: Install atari dependencies | |
run: poetry install -E "pytest atari" | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Run atari tests | |
run: poetry run pytest tests/test_atari_multigpu.py | |
test-pettingzoo-envs: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.8", "3.9", "3.10"] | |
poetry-version: ["1.7"] | |
os: [ubuntu-22.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Run image | |
uses: abatilo/actions-poetry@v2.0.0 | |
with: | |
poetry-version: ${{ matrix.poetry-version }} | |
# pettingzoo tests | |
- name: Install pettingzoo dependencies | |
run: poetry install -E "pytest pettingzoo atari" | |
- name: Downgrade setuptools | |
run: poetry run pip install setuptools==59.5.0 | |
- name: Install ROMs | |
run: poetry run AutoROM --accept-license | |
- name: Run pettingzoo tests | |
run: poetry run pytest tests/test_pettingzoo_ma_atari.py |