lightning/pytorch_lightning/__init__.py

from .models.trainer import Trainer
from .root_module.root_module import LightningModule
from .root_module.decorators import data_loader

__all__ = [
    'Trainer',
    'LightningModule',
    'data_loader',
]