ecc8b0b34138fa480e1b1876fd2695c07be9b159,simple_rl/tasks/navigation/NavigationMDP.py,NavigationMDP,reset_goals,#NavigationMDP#,179

Before Change

        self.cell_rewards = self.cell_rewards_wo_goals.copy()

        // Update goals and their rewards
        self.all_goals_state_features_entangled = len(np.unique(goal_colors)) == 1
        for goal_idx, goal_loc in enumerate(self.goal_locs):
            goal_r, goal_c = self._xy_to_rowcol(goal_loc[0], goal_loc[1])
            if self.all_goals_state_features_entangled:

After Change

        self.goal_cell_locs = goal_cell_locs
        self.goal_cell_rewards = goal_cell_rewards
        self.goal_cell_types = goal_types
        self.goal_cell_ids = list(range(self.living_cell_ids[-1] + 1,
                                        self.living_cell_ids[-1] + 1 + len(
                                            self.goal_cell_locs)))
        // Reset goal xy to idx dict

Instances