Update the batch building to reflect latest changes to BenchmarkSpecification.
Update the batch building to reflect latest changes to BenchmarkSpecification.

Updates the creation of batches to only include datasets that are specified to be used for each particular split.

PiperOrigin-RevId: 257186478

google-research/meta-dataset