diff --git a/test/common/torchtext_test_case.py b/test/common/torchtext_test_case.py index 322eb32292..c456cd705b 100644 --- a/test/common/torchtext_test_case.py +++ b/test/common/torchtext_test_case.py @@ -108,7 +108,7 @@ def make_mock_dataset(self, num_examples=30, num_labels=3): texts = [str(i) for i in range(num_examples)] labels = list(range(num_labels)) * num_repetitions - labels = [str(l) for l in labels[:num_examples]] + labels = [str(line) for line in labels[:num_examples]] dict_dataset = [ {'text': t, 'label': l} for t, l in zip(texts, labels) diff --git a/torchtext/datasets/translation.py b/torchtext/datasets/translation.py index 058022999f..ac1b2f322a 100644 --- a/torchtext/datasets/translation.py +++ b/torchtext/datasets/translation.py @@ -178,9 +178,9 @@ def clean(path): f_txt = f_orig.replace('.tags', '') with codecs.open(f_txt, mode='w', encoding='utf-8') as fd_txt, \ io.open(f_orig, mode='r', encoding='utf-8') as fd_orig: - for l in fd_orig: - if not any(tag in l for tag in xml_tags): - fd_txt.write(l.strip() + '\n') + for line in fd_orig: + if not any(tag in line for tag in xml_tags): + fd_txt.write(line.strip() + '\n') class WMT14(TranslationDataset):