code: def on_batch_end(self):
          
          for callback in self.callbacks:
              callback.on_batch_end(self)
  
          # Even with all the experience replay, we count the single rollout as a single batch
          self.epoch_info.result_accumulator.calculate(self)
docs: Finalize batch processing