Spaces:
Sleeping
Sleeping
File size: 496 Bytes
3f7c971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import os
from .base_atari_env import BaseAtariEnv, base_env_wrapper_fn, parallel_wrapper_fn
def raw_env(num_players=2, **kwargs):
assert num_players == 2 or num_players == 4, "pong only supports 2 or 4 players"
mode_mapping = {2: 45, 4: 49}
mode = mode_mapping[num_players]
return BaseAtariEnv(game="pong", num_players=num_players, mode_num=mode, env_name=os.path.basename(__file__)[:-3], **kwargs)
env = base_env_wrapper_fn(raw_env)
parallel_env = parallel_wrapper_fn(env)
|