[Tests] Update tests after torchrl bugfix

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Oct 6, 2023 · b32c0a2 · b32c0a2
1 parent ee440e0
commit b32c0a2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/test/test_pettingzoo.py b/test/test_pettingzoo.py
@@ -5,7 +5,6 @@
 from benchmarl.algorithms import (
     algorithm_config_registry,
     IppoConfig,
-    IsacConfig,
     MaddpgConfig,
     MasacConfig,
     QmixConfig,
@@ -94,7 +93,7 @@ def test_reloading_trainer(
     ):
         experiment_config.prefer_continuous_actions = prefer_continuous
         algo_config = algo_config.get_from_yaml()
-        if isinstance(algo_config, (VdnConfig, IsacConfig, MasacConfig)):
+        if isinstance(algo_config, VdnConfig):
             # There are some bugs currently in TorchRL
             return
         ExperimentUtils.check_experiment_loading(

diff --git a/test/test_vmas.py b/test/test_vmas.py
@@ -5,7 +5,6 @@
 from benchmarl.algorithms import (
     algorithm_config_registry,
     IppoConfig,
-    IsacConfig,
     MaddpgConfig,
     MappoConfig,
     MasacConfig,
@@ -82,7 +81,7 @@ def test_reloading_trainer(
         mlp_sequence_config,
     ):
         algo_config = algo_config.get_from_yaml()
-        if isinstance(algo_config, (VdnConfig, IsacConfig, MasacConfig)):
+        if isinstance(algo_config, VdnConfig):
             # There are some bugs currently in TorchRL
             return
         ExperimentUtils.check_experiment_loading(