def METHOD_NAME(self) -> int:
    data_source = getattr(self._dataset, "_data_source")
    if not hasattr(data_source, "_batch_size"):
        return 1
    METHOD_NAME = getattr(data_source, "_batch_size")
    try:
        if isinstance(METHOD_NAME, tf.Tensor):
            METHOD_NAME = METHOD_NAME.numpy()
        METHOD_NAME = int(METHOD_NAME)
    except:  # pylint: disable=W0702
        METHOD_NAME = 1
    return METHOD_NAME