from abc import ABC, abstractmethod class TrainDataloaderVariations(ABC): @abstractmethod def dataloader(self, train: bool): """placeholder""" def train_dataloader(self): return self.dataloader(train=True)