def test_smoke():
        """A simple test that the class can be instantiated."""
        dm = TemplateData(batch_size=2)
        assert dm is not None
A simple test that the class can be instantiated.
tests/template/classification/test_data.py