missed a not

7021ac1c · Taylor Robie · f1efaf83 · 7021ac1c
Commit 7021ac1c authored Jan 08, 2019 by Taylor Robie
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/recommendation/data_pipeline.py official/recommendation/data_pipeline.py +1 -1

No files found.
--- a/official/recommendation/data_pipeline.py
+++ b/official/recommendation/data_pipeline.py
@@ -378,7 +378,7 @@ class BaseDataConstructor(threading.Thread):
    self._current_epoch_order = np.empty(shape=(0,))
    self._shuffle_iterator = None

-    self._shuffle_with_forkpool = stream_files
+    self._shuffle_with_forkpool = not stream_files
    if stream_files:
      self._shard_root = tempfile.mkdtemp(prefix="ncf_")
      atexit.register(tf.gfile.DeleteRecursively, dirname=self._shard_root)