From 24c0f2026885d78fae097736587bc47486507aab Mon Sep 17 00:00:00 2001 From: Matej Aleksandrov Date: Mon, 26 Oct 2026 03:43:17 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 1007157877 --- tensorflow_datasets/core/writer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_datasets/core/writer.py b/tensorflow_datasets/core/writer.py index 8b8f8eacaa1..0b89b40acf7 100644 --- a/tensorflow_datasets/core/writer.py +++ b/tensorflow_datasets/core/writer.py @@ -716,7 +716,7 @@ def write_from_pcollection(self, examples_pcollection): # (_ShardInfo) | "CollectShardInfo" >> beam.transforms.combiners.ToList() # [_ShardInfo] - | "CalculateSplitInfo" >> beam.ParDo(self._store_split_info) + | "CalculateSplitInfo" >> beam.Map(self._store_split_info) ) def finalize(self) -> tuple[list[int], int]: