Skip to content

Commit caaa8e3

Browse files
authored
Remove redundant dataname in test_shuffle_shard_wrapper (#1733)
1 parent 996d9f8 commit caaa8e3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/datasets/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,8 @@
1010
class TestShuffleShardDatasetWrapper(TorchtextTestCase):
1111
# Note that for order i.e shuffle before sharding, TorchData will provide linter warning
1212
# 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):
13+
@parameterized.expand([(f,) for f in DATASETS.values()])
14+
def test_shuffle_shard_wrapper(self, dataset_fn):
1515
dp = dataset_fn()
1616
if type(dp) == tuple:
1717
dp = list(dp)

0 commit comments

Comments
 (0)