lightning/tests/base/eval_model_test_dataloaders.py

12 lines
237 B
Python

from abc import ABC, abstractmethod
class TestDataloaderVariations(ABC):
@abstractmethod
def dataloader(self, train: bool):
"""placeholder"""
def test_dataloader(self):
return self.dataloader(train=False)