Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 715fcde

Browse files
committed
try to fix style.
1 parent aa6af76 commit 715fcde

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

torchtext/datasets/iwslt2016.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,10 @@
55
from torchdata.datapipes.iter import FileOpener, GDriveReader, IterableWrapper, FileLister
66

77
import os
8-
from torchtext.utils import (download_from_url, extract_archive)
98
from torchtext.data.datasets_utils import (
10-
_RawTextIterableDataset,
119
_wrap_split_argument,
1210
_clean_xml_file,
1311
_clean_tags_file,
14-
_read_text_iterator,
1512
)
1613
from torchtext.data.datasets_utils import _create_dataset_directory
1714

@@ -160,7 +157,7 @@ def _construct_filepaths(paths, src_filename, tgt_filename):
160157

161158
@_create_dataset_directory(dataset_name=DATASET_NAME)
162159
@_wrap_split_argument(("train", "valid", "test"))
163-
def IWSLT2016(root = '.data', split=('train', 'valid', 'test'), language_pair=('de', 'en'), valid_set='tst2013', test_set='tst2014'):
160+
def IWSLT2016(root='.data', split=('train', 'valid', 'test'), language_pair=('de', 'en'), valid_set='tst2013', test_set='tst2014'):
164161
"""IWSLT2016 dataset
165162
166163
The available datasets include following:
@@ -198,12 +195,6 @@ def IWSLT2016(root = '.data', split=('train', 'valid', 'test'), language_pair=('
198195
>>> src_sentence, tgt_sentence = next(train_iter)
199196
200197
"""
201-
num_lines_set_identifier = {
202-
'train': 'train',
203-
'valid': valid_set,
204-
'test': test_set
205-
}
206-
207198
if not is_module_available("torchdata"):
208199
raise ModuleNotFoundError("Package `torchdata` not found. Please install following instructions at `https://github.com/pytorch/data`")
209200

0 commit comments

Comments
 (0)