code: def on_fit_end(self):
      assert self == self.trainer.lightning_module
  
file_path: pytorch-lightning/tests/tests_pytorch/trainer/properties/test_get_model.py
imports: import pytest
  from lightning.pytorch import Trainer
  from lightning.pytorch.demos.boring_classes import BoringModel
  from tests_pytorch.helpers.runif import RunIf
label_desc: Use image node_id 2 for calling the TrainerGetModel obj's underlying member method code with example usage: obj.on_fit_end() without return types