lightning/tests/base/eval_model_train_dataloaders.py

from abc import ABC, abstractmethod


class TrainDataloaderVariations(ABC):

    @abstractmethod
    def dataloader(self, train: bool):
        """placeholder"""

    def train_dataloader(self):
        return self.dataloader(train=True)