diff --git a/envs/ma_gym/__init__.py b/envs/ma_gym/__init__.py index 293e16a..aaa72e7 100755 --- a/envs/ma_gym/__init__.py +++ b/envs/ma_gym/__init__.py @@ -7,7 +7,7 @@ # Register openai's environments as multi agent # This should be done before registering new environments -env_specs = [env_spec for env_spec in envs.registry.all() if 'gym.envs' in env_spec.entry_point] +env_specs = [env_spec for env_spec in envs.registry.values() if 'gym.envs' in env_spec.entry_point] for spec in env_specs: register( id='ma_' + spec.id,