コード例 #1
0
ファイル: test_monitoring.py プロジェクト: zmoon111/universe
def test_vnc_monitoring():
    with helpers.tempdir() as temp:
        env = gym.make('gym-core.Pong-v3')
        env = wrappers.GymCoreAction(env)
        env = wrappers.Monitor(temp)(env)

        env.configure(remotes=2)
        env.reset()
        for i in range(2):
            env.step([0, 0])
        env.close()

        results = gym.monitoring.load_results(temp)
        assert results['env_info']['env_id'] == 'gym-core.Pong-v3'
コード例 #2
0
def atari_vnc_wrapper(env):
    env = wrappers.Vision(env)
    env = wrappers.GymCoreAction(env)
    return env