examples/research_projects/codeparrot/examples/train_complexity_predictor.py/CustomCallback/__init__
class CustomCallback:    def __init__(self, trainer) -> None:
        super().__init__()
        self._trainer = trainer