keras/src/trainers/data_adapters/py_dataset_adapter.py/PyDataset/__getitem__
class PyDataset:    def __getitem__(self, index):
        """Gets batch at position `index`.

        Args:
            index: position of the batch in the PyDataset.

        Returns:
            A batch
        """
        raise NotImplementedError