code: def on_train_epoch_end(self):
      _ = torch.stack(self.training_step_outputs).mean()
      self.training_step_outputs.clear()
  
file_path: wandb/tests/functional_tests/t0_main/lightning/pl_base.py
imports: import torch
  from lightning import LightningModule
  from torch.utils.data import Dataset
label_desc: Use image node_id 6 for calling the BoringModel obj's underlying member method code with example usage: obj.on_train_epoch_end() without return types