19 lines
467 B
Python
19 lines
467 B
Python
|
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,
|
||
|
|
)
|