I\'m using the minigrid environment with the torch-rl extension. I\'m trying to get the agent to track a calibration icon for set number of steps, then move off to do other