diff --git a/.github/workflows/torchrl_stable_tests.yml b/.github/workflows/torchrl_stable_tests.yml index 3ca7e448..9ce35ec1 100644 --- a/.github/workflows/torchrl_stable_tests.yml +++ b/.github/workflows/torchrl_stable_tests.yml @@ -36,6 +36,9 @@ jobs: - name: Install pettingzoo run: | bash .github/unittest/install_pettingzoo.sh + - name: Install meltingpot + run: | + bash .github/unittest/install_meltingpot.sh - name: Tests run: | xvfb-run -s "-screen 0 1024x768x24" pytest test/test_algorithm.py test/test_models.py test/test_task.py test/test_vmas.py test/test_pettingzoo.py test/test_meltingpot.py --doctest-modules --junitxml=junit/test-results.xml --cov=. --cov-report=xml --cov-report=html diff --git a/test/test_meltingpot.py b/test/test_meltingpot.py index db6a5474..2e7b20f9 100644 --- a/test/test_meltingpot.py +++ b/test/test_meltingpot.py @@ -59,7 +59,7 @@ def test_all_algos( experiment.run() @pytest.mark.parametrize("algo_config", [MasacConfig]) - @pytest.mark.parametrize("task", _get_unique_envs(list(MeltingPotTask))) + @pytest.mark.parametrize("task", _get_unique_envs(list(MeltingPotTask))[:10]) def test_all_tasks( self, algo_config: AlgorithmConfig,