Reshuffle smart buildings dataset before writing

PiperOrigin-RevId: 702259211
tensorflow · Dec 3, 2024 · 0be515f · 0be515f
1 parent 2d75769
commit 0be515f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_datasets/datasets/smart_buildings/smart_buildings_dataset_builder.py b/tensorflow_datasets/datasets/smart_buildings/smart_buildings_dataset_builder.py
@@ -198,6 +198,7 @@ def _generate_examples(self, path: epath.Path, year: int, pipeline):
         pipeline
         | f'CreateDates_{year}' >> beam.Create(all_dates)
         | f'ProcessDate_{year}' >> beam.FlatMap(process_date, path=path)
+        | f'Reshuffle_{year}' >> beam.Reshuffle()
     )