Update tfds.load() callers to specify shuffle_files=True when nec…

…essary. PiperOrigin-RevId: 264401453
gridl · Aug 20, 2019 · 98a2052 · 98a2052
1 parent 8f35157
commit 98a2052
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tensorflow_datasets/core/dataset_builder_test.py b/tensorflow_datasets/core/dataset_builder_test.py
@@ -125,7 +125,7 @@ def test_determinism(self):
           name="dummy_dataset_shared_generator",
           data_dir=tmp_dir,
           split=splits_lib.Split.TRAIN,
-          as_dataset_kwargs=dict(shuffle_files=False))
+          shuffle_files=False)
       ds_values = list(dataset_utils.as_numpy(ds))
 
       # Ensure determinism. If this test fail, this mean that numpy random
@@ -152,7 +152,7 @@ def test_multi_split(self):
           name="dummy_dataset_shared_generator",
           data_dir=tmp_dir,
           split=["train", "test"],
-          as_dataset_kwargs=dict(shuffle_files=False))
+          shuffle_files=False)
 
       data = list(dataset_utils.as_numpy(ds_train))
       self.assertEqual(20, len(data))

diff --git a/tensorflow_datasets/video/moving_sequence.py b/tensorflow_datasets/video/moving_sequence.py
@@ -146,7 +146,8 @@ def animate(sequence):
 
 
   tf.enable_eager_execution()
-  mnist_ds = tfds.load("mnist", split=tfds.Split.TRAIN, as_supervised=True)
+  mnist_ds = tfds.load("mnist", split=tfds.Split.TRAIN, as_supervised=True,
+                       shuffle_files=True)
   mnist_ds = mnist_ds.repeat().shuffle(1024)
 
   def map_fn(image, label):