"""Module for PyTorch Lightning callbacks."""