We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 996d9f8 commit caaa8e3Copy full SHA for caaa8e3
test/datasets/common.py
@@ -10,8 +10,8 @@
10
class TestShuffleShardDatasetWrapper(TorchtextTestCase):
11
# Note that for order i.e shuffle before sharding, TorchData will provide linter warning
12
# Modify this test when linter warning is available
13
- @parameterized.expand(list(DATASETS.items()))
14
- def test_shuffle_shard_wrapper(self, dataset_name, dataset_fn):
+ @parameterized.expand([(f,) for f in DATASETS.values()])
+ def test_shuffle_shard_wrapper(self, dataset_fn):
15
dp = dataset_fn()
16
if type(dp) == tuple:
17
dp = list(dp)
0 commit comments