244aafdcf89ae814975c8c4e3faf0bd4995c7878,rllib/utils/exploration/tests/test_curiosity.py,OneHotWrapper,observation,#OneHotWrapper#Any#,29

Before Change

        // Are we carrying the key?
        if self.carrying is not None:
            print("Carrying KEY!!")

        self.x_positions.append(self.agent_pos[0])
        self.y_positions.append(self.agent_pos[1])

After Change

    def observation(self, obs):
        // Debug output: max-x/y positions to watch exploration progress.
        if self.step_count == 0:
            for _ in range(self.framestack):
                self.frame_buffer.append(np.zeros((self.single_frame_dim, )))
            if self.vector_index == 0:
                if self.x_positions:
                    max_diff = max(
                        np.sqrt((np.array(self.x_positions) - self.init_x)**2 +

Instances