def METHOD_NAME(self, samples, **extra_args):
    # For now only supports datasets with same underlying collater implementations
    if hasattr(self.datasets[0], "collater"):
        return self.datasets[0].METHOD_NAME(samples, **extra_args)
    else:
        return default_collate(samples, **extra_args)