lightning/pytorch_lightning/__init__.py

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

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