Skip to content

Commit b54b9c2

Browse files
yangustc07copybara-github
authored andcommitted
#tf-data Migrate tf.data.experimental.RandomDataset to tf.data.Dataset.random.
PiperOrigin-RevId: 731073912
1 parent 3902386 commit b54b9c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_examples/lite/model_maker/core/task/model_spec/audio_spec_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def fn(value):
3838

3939
return fn
4040

41-
wav_ds = tf.data.experimental.RandomDataset(seed=seed).take(total_samples)
41+
wav_ds = tf.data.Dataset.random(seed=seed).take(total_samples)
4242
wav_ds = wav_ds.map(fill_shape([
4343
spec.target_sample_rate,
4444
]))

0 commit comments

Comments
 (0)