[CI] Reduce metlingpot test time (#81)

facebookresearch · Apr 26, 2024 · e6d46be · e6d46be
1 parent 729aa2a
commit e6d46be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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,