Spaces:
Sleeping
Sleeping
File size: 597 Bytes
3f7c971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import os
from .base_atari_env import BaseAtariEnv, base_env_wrapper_fn, parallel_wrapper_fn
avaliable_versions = {
"bi-plane": 15,
"jet": 21,
}
def raw_env(game_version="bi-plane", guided_missile=True, **kwargs):
assert game_version in avaliable_versions, "game_version must be either 'jet' or 'bi-plane'"
mode = avaliable_versions[game_version] + (0 if guided_missile else 1)
return BaseAtariEnv(game="combat", 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)
|