ray.rllib.offline.offline_prelearner.OfflinePreLearner._map_sample_batch_to_episode

OfflinePreLearner._map_sample_batch_to_episode(batch: Dict[str, list | numpy.ndarray], to_numpy: bool = False, **kwargs: Dict[str, Any]) → Dict[str, List[SingleAgentEpisode | MultiAgentEpisode]]

Maps an old stack SampleBatch to new stack episodes.