Source code for mushroom_rl.utils.callbacks.collect_dataset

from mushroom_rl.utils.callbacks.callback import Callback


[docs]class CollectDataset(Callback): """ This callback can be used to collect samples during the learning of the agent. """
[docs] def __call__(self, dataset): self._data_list += dataset