def reset(self):
        """
        **Overview**:
            Reset environment and process obs.
        """
        # Reset environment.
        obs = self.env.reset()
        # Add optical flow channels into obs.
        state = self._process_obs(obs)
        # Update last frame.
        self.last_frame = obs
        return state

        **Overview**:
            Reset environment and process obs.
        
chapter3_obs/mario_wrapper.py