def action_from_keys(self, keys): if keys[pygame.K_LEFT]: return np.asarray([-1]) elif keys[pygame.K_RIGHT]: return np.asarray([+1]) else: return np.asarray([0])