ray.rllib.evaluation.sampler.SamplerInput.get_extra_batches

abstract SamplerInput.get_extra_batches() → List[SampleBatch | MultiAgentBatch]

Returns list of extra batches since the last call to this method.

The list will contain all SampleBatches or MultiAgentBatches that the user has provided thus-far. Users can add these “extra batches” to an episode by calling the episode’s add_extra_batch([SampleBatchType]) method. This can be done from inside an overridden Policy.compute_actions_from_input_dict(..., episodes) or from a custom callback’s on_episode_[start|step|end]() methods.

Returns:

List of SamplesBatches or MultiAgentBatches provided thus-far by the user since the last call to this method.