def test_sharding(self):
    for i in range(3):
        self.verify_task_matches_fake_datasets('cached_task', use_cached=False, num_shards=i, token_preprocessed=False)
        self.verify_task_matches_fake_datasets('cached_task', use_cached=True, num_shards=i, token_preprocessed=False)
def test_sharding(self):
    for i in range(3):
        self.verify_task_matches_fake_datasets('cached_task', use_cached=False, num_shards=i, token_preprocessed=False)
        self.verify_task_matches_fake_datasets('cached_task', use_cached=True, num_shards=i, token_preprocessed=False)