Spaces:
Sleeping
Sleeping
| from mlagents_envs.registry import default_registry | |
| from mlagents_envs.envs.pettingzoo_env_factory import logger, PettingZooEnvFactory | |
| # Register each environment in default_registry as a PettingZooEnv | |
| for key in default_registry: | |
| env_name = key | |
| if key[0].isdigit(): | |
| env_name = key.replace("3", "Three") | |
| if not env_name.isidentifier(): | |
| logger.warning( | |
| f"Environment id {env_name} can not be registered since it is" | |
| f"not a valid identifier name." | |
| ) | |
| continue | |
| locals()[env_name] = PettingZooEnvFactory(key) | |