"""PyTorch Lightning models modules."""