amend

facebookresearch · Apr 18, 2024 · daf8add · daf8add
1 parent d058702
commit daf8add
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/.github/workflows/torchrl_stable_tests.yml b/.github/workflows/torchrl_stable_tests.yml
@@ -38,4 +38,4 @@ jobs:
         bash .github/unittest/install_pettingzoo.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 --doctest-modules --junitxml=junit/test-results.xml --cov=. --cov-report=xml --cov-report=html
+        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
@@ -5,7 +5,9 @@
 #
 
 
+import packaging
 import pytest
+import torchrl
 
 from benchmarl.algorithms import (
     algorithm_config_registry,
@@ -23,6 +25,10 @@
 
 
 @pytest.mark.skipif(not _has_meltingpot, reason="Meltingpot not found")
+@pytest.mark.skipif(
+    not packaging.version.parse(torchrl.__version__).base_version <= "0.3.1",
+    reason="TorchRL <= 0.3.1 does nto support meltingpot",
+)
 class TestMeltingPot:
     @pytest.mark.parametrize("algo_config", algorithm_config_registry.values())
     @pytest.mark.parametrize("task", [MeltingPotTask.COMMONS_HARVEST__OPEN])