from lightning.pytorch.utilities.types import EVAL_DATALOADERS
from .base_data_module import BaseDataModule


class ValDataModule(BaseDataModule):
    def val_dataloader(self) -> EVAL_DATALOADERS:
        return [
            self._dataloader(split) for split in self.exp_cfg.validation.dataset_splits
        ]