How severe does this issue affect your experience of using Ray?
- Low: It annoys or frustrates me for a moment.
I am trying to create a custom callback that tracks a certain element from the observation vector at the end of an episode. Episode has a method to grab the last observation but it is removed in V2. I wondered if there was a clean way to do this or if I would have to grab from the AgentCollector.