amend

facebookresearch · Apr 26, 2024 · dfb1ae9 · dfb1ae9
1 parent 6cd9b49
commit dfb1ae9
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/test/test_meltingpot.py b/test/test_meltingpot.py
@@ -5,9 +5,7 @@
 #
 
 
-import packaging
 import pytest
-import torchrl
 
 from benchmarl.algorithms import (
     algorithm_config_registry,
@@ -35,10 +33,6 @@ def _get_unique_envs(names):
 
 
 @pytest.mark.skipif(not _has_meltingpot, reason="Meltingpot not found")
-@pytest.mark.skipif(
-    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])

diff --git a/test/test_models.py b/test/test_models.py
@@ -6,10 +6,8 @@
 
 from typing import List
 
-import packaging
 import pytest
 import torch
-import torchrl
 
 from benchmarl.hydra_config import load_model_config_from_hydra
 from benchmarl.models import model_config_registry
@@ -79,11 +77,6 @@ def test_models_forward_shape(
         pytest.skip()  # this combination should never happen
     if ("gnn" in model_name) and centralised:
         pytest.skip("gnn model is always decentralized")
-    if (
-        packaging.version.parse(torchrl.__version__).base_version <= "0.3.1"
-        and "cnn" in model_name
-    ):
-        pytest.skip("TorchRL <= 0.3.1 does not support MultiAgentCNN")
 
     torch.manual_seed(0)