fix flake8 errors

Signed-off-by: Eli Uriegas <eliuriegas@fb.com> (cherry picked from commit 100a855) Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
pytorch · Feb 18, 2021 · b48fa1b · b48fa1b
1 parent 5cfda00
commit b48fa1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)