Spaces:
Sleeping
Sleeping
File size: 989 Bytes
3f7c971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os
import warnings
from .base_atari_env import BaseAtariEnv, base_env_wrapper_fn, parallel_wrapper_fn
avaliable_versions = {
"robbers": 2,
"race": 1,
"capture": 12,
}
def raw_env(game_version="robbers", visibilty_level=0, **kwargs):
if game_version == "robbers" and visibilty_level == 0:
warnings.warn("maze_craze has different versions of the game via the `game_version` argument, consider overriding.")
assert game_version in avaliable_versions, f"`game_version` parameter must be one of {avaliable_versions.keys()}"
assert 0 <= visibilty_level < 4, "visibility level must be between 0 and 4, where 0 is 100% visibility and 3 is 0% visibility"
base_mode = (avaliable_versions[game_version] - 1) * 4
mode = base_mode + visibilty_level
return BaseAtariEnv(game="maze_craze", num_players=2, mode_num=mode, env_name=os.path.basename(__file__)[:-3], **kwargs)
env = base_env_wrapper_fn(raw_env)
parallel_env = parallel_wrapper_fn(env)
|