update hint_shuffling (#5693)

* update hint_shuffling * trigger CI

update hint_shuffling (#5693)
* update hint_shuffling * trigger CI
aa211974 · Philip Meier · GitHub · f1172393 · aa211974
Unverified Commit aa211974 authored Mar 29, 2022 by Philip Meier Committed by GitHub Mar 29, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

torchvision/prototype/datasets/utils/_internal.py torchvision/prototype/datasets/utils/_internal.py +2 -1

No files found.
--- a/torchvision/prototype/datasets/utils/_internal.py
+++ b/torchvision/prototype/datasets/utils/_internal.py
@@ -34,6 +34,7 @@ __all__ = [
    "path_comparator",
    "read_flo",
    "hint_sharding",
+    "hint_shuffling",
 ]

 K = TypeVar("K")
@@ -196,4 +197,4 @@ def hint_sharding(datapipe: IterDataPipe) -> ShardingFilter:


 def hint_shuffling(datapipe: IterDataPipe[D]) -> Shuffler[D]:
-    return Shuffler(datapipe, default=False, buffer_size=INFINITE_BUFFER_SIZE)
+    return Shuffler(datapipe, buffer_size=INFINITE_BUFFER_SIZE).set_shuffle(False)