lightning/tests/tests_pytorch/helpers/__init__.py

from lightning.pytorch.demos.boring_classes import (  # noqa: F401
    BoringDataModule,
    BoringModel,
    ManualOptimBoringModel,
    RandomDataset,
)

from tests_pytorch.helpers.datasets import TrialMNIST  # noqa: F401