From cf7e4a62f6f44d1fb993cc00f98194b91b37a955 Mon Sep 17 00:00:00 2001 From: Parmeet Singh Bhatia Date: Thu, 19 May 2022 08:39:51 -0400 Subject: [PATCH] Remove redundant dataname in test_shuffle_shard_wrapper --- test/datasets/common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/datasets/common.py b/test/datasets/common.py index 81edc565bf..f9ead3d3b3 100644 --- a/test/datasets/common.py +++ b/test/datasets/common.py @@ -10,8 +10,8 @@ class TestShuffleShardDatasetWrapper(TorchtextTestCase): # Note that for order i.e shuffle before sharding, TorchData will provide linter warning # Modify this test when linter warning is available - @parameterized.expand(list(DATASETS.items())) - 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): dp = dataset_fn() if type(dp) == tuple: dp = list(dp)