This repository has been archived by the owner on Apr 6, 2018. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 956
/
__init__.py
59 lines (48 loc) · 2.16 KB
/
__init__.py
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import gym
import universe.wrappers.experimental
from universe import envs, spaces
from universe.wrappers import gym_core_sync
from universe.wrappers.blocking_reset import BlockingReset
from universe.wrappers.diagnostics import Diagnostics
from universe.wrappers.gym_core import GymCoreAction, GymCoreObservation, CropAtari
from universe.wrappers.joint import Joint
from universe.wrappers.logger import Logger
from universe.wrappers.monitoring import Monitor
from universe.wrappers.multiprocessing_env import WrappedMultiprocessingEnv, EpisodeID
from universe.wrappers.recording import Recording
from universe.wrappers.render import Render
from universe.wrappers.throttle import Throttle
from universe.wrappers.time_limit import TimeLimit
from universe.wrappers.timer import Timer
from universe.wrappers.vectorize import Vectorize, Unvectorize, WeakUnvectorize
from universe.wrappers.vision import Vision
def wrap(env):
return Timer(Render(Throttle(env)))
def WrappedVNCEnv():
return wrap(envs.VNCEnv())
def WrappedGymCoreEnv(gym_core_id, fps=None, rewarder_observation=False):
# Don't need to store the ID on the instance; it'll be retrieved
# directly from the spec
env = wrap(envs.VNCEnv(fps=fps))
if rewarder_observation:
env = GymCoreObservation(env, gym_core_id=gym_core_id)
return env
def WrappedGymCoreSyncEnv(gym_core_id, fps=60, rewarder_observation=False):
spec = gym.spec(gym_core_id)
env = gym_core_sync.GymCoreSync(BlockingReset(wrap(envs.VNCEnv(fps=fps))))
if rewarder_observation:
env = GymCoreObservation(env, gym_core_id=gym_core_id)
elif spec._entry_point.startswith('gym.envs.atari:'):
env = CropAtari(env)
return env
def WrappedFlashgamesEnv():
keysym = spaces.KeyEvent.by_name('`').key
return wrap(envs.VNCEnv(probe_key=keysym))
def WrappedInternetEnv(*args, **kwargs):
return wrap(envs.InternetEnv(*args, **kwargs))
def WrappedStarCraftEnv(*args, **kwargs):
return wrap(envs.StarCraftEnv(*args, **kwargs))
def WrappedGTAVEnv(*args, **kwargs):
return wrap(envs.GTAVEnv(*args, **kwargs))
def WrappedWorldOfGooEnv(*args, **kwargs):
return wrap(envs.WorldOfGooEnv(*args, **kwargs))