diff --git a/test/test_pettingzoo.py b/test/test_pettingzoo.py index 1bc8fd27..dcddb654 100644 --- 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 index 2d427b5e..b0f8eed0 100644 --- 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(