Migrate tf.data.experimental.RandomDataset to `tf.data.Dataset.rand…

…om`. PiperOrigin-RevId: 731366527
google · Feb 26, 2025 · bb079d8 · bb079d8
1 parent f9b84ed
commit bb079d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/seqio/utils.py b/seqio/utils.py
@@ -1472,7 +1472,7 @@ def __call__(self, dataset: tf.data.Dataset, *args, **kwargs):
       random_ds_seeds = tuple(tuple(s) for s in random_ds_seeds)
       _NEXT_MAP_SEED += 2 * self.num_seeds
     seed_datasets = tf.nest.map_structure(
-        tf.data.experimental.RandomDataset, random_ds_seeds
+        tf.data.Dataset.random, random_ds_seeds
     )
 
     def map_fn(element, seeds):