def setUp(self): self._env = test_environment_decorator.TestEnvironmentDecorator( deepmind_lab.Lab( 'seekavoid_arena_01', ['VEL.ROT', 'VEL.TRANS', 'DEBUG.POS.ROT', 'DEBUG.POS.TRANS'])) self._controller = game_controller.GameController(self._env) self._env.reset()
def __init__(self, env): """Initialize the controller.""" self._env = env self._controller = game_controller.GameController(env) self._maze = None self._pickup_locations = None