bmstu-marl/multiagent/__init__.py

19 lines
467 B
Python
Raw Normal View History

from gym.envs.registration import register
# Multiagent envs
# ----------------------------------------
register(
id='MultiagentSimple-v0',
entry_point='multiagent.envs:SimpleEnv',
# FIXME(cathywu) currently has to be exactly max_path_length parameters in
# rllab run script
max_episode_steps=100,
)
register(
id='MultiagentSimpleSpeakerListener-v0',
entry_point='multiagent.envs:SimpleSpeakerListenerEnv',
max_episode_steps=100,
)