From b48fa1b1501e2d359a401178a8a210467198e375 Mon Sep 17 00:00:00 2001 From: Eli Uriegas Date: Tue, 1 Dec 2020 14:11:11 -0800 Subject: [PATCH] fix flake8 errors Signed-off-by: Eli Uriegas (cherry picked from commit 100a855d3cbcc1505d89077bcb6d23d17c4c4c2b) Signed-off-by: Eli Uriegas --- test/common/torchtext_test_case.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)