Returns the pickable state of an episode.
The data in the episode is stored into a dictionary. Note that episodes
can also be generated from states (see SingleAgentEpisode.from_state()).
- Returns:
A dict containing all the data from the episode.
- Return type:
Dict[str, Any]