Spaces:
Running
Running
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) | |