def reset(self, obstacle_count=0): self.iterations = 0 self.bot.reset() self.obstacles = Obstacle.make_obstacles(obstacle_count) obs = self.bot.get_observation(self.obstacles) self.min_distance = min(obs) # for use in first call to step() return np.argmin(obs) # convert to discrete observation
def reset(self, obstacle_count=0, obstacles=None): self.iterations = 0 self.bot.reset() if obstacles is None: self.obstacles = Obstacle.make_obstacles(obstacle_count) else: self.obstacles = obstacles obs = self.bot.get_observation(self.obstacles) self.min_distance = min(obs) # for use in first call to step() return self.get_discrete_observation(obs) # return discrete observation
def reset(self): self.x = 0 self.y = 0 self.heading = 0 # points directly to the right --> self.n_heading = 0 # eliminates rounding error self.obstacles = Obstacle.make_obstacles(self.obstacle_count)