standalone.py

#!/usr/bin/env python2

from __future__ import division, print_function

import gym

from gym_duckietown.envs import DuckietownEnv
import pyglet

def main():

    env = gym.make('Duckietown-v0')
    env.reset()

    env.render()
    @env.window.event
    def on_key_press(symbol, modifiers):
        from pyglet.window import key

        action = None
        if symbol == key.LEFT:
            print('left')
            action = (-1, 1)
        elif symbol == key.RIGHT:
            print('right')
            action = (1, -1)
        elif symbol == key.UP:
            print('forward')
            action = (1, 1)
        elif symbol == key.DOWN:
            print('back')
            action = (-1, -1)
        elif symbol == key.SLASH:
            print('RESET')
            action = (0,0)
            env.reset()
        else:
            return

        if action != (0,0):
            obs, reward, done, info = env.step(action)

            print(reward, done)

            env.render()

            if done:
                print('done!')
                env.reset()

    # Enter main event loop
    pyglet.app.run()

if __name__ == "__main__":
    main()