guides/training_with_built_in_methods.py/ExamplePyDataset/__init__
class ExamplePyDataset:    def __init__(self, x, y, batch_size, **kwargs):
        super().__init__(**kwargs)
        self.x = x
        self.y = y
        self.batch_size = batch_size