def METHOD_NAME(self):
    if isinstance(self.dataset.METHOD_NAME, list):
        return [s[self._cur_indices.array] for s in self.dataset.METHOD_NAME]
    return self.dataset.METHOD_NAME[self._cur_indices.array]