return for SQuAD [BLACKED]

d4614729 · Lysandre · f24a228a · d4614729 · d4614729
Commit d4614729 authored Dec 13, 2019 by Lysandre
Showing with 172 additions and 110 deletions

transformers/data/processors/glue.py transformers/data/processors/glue.py +1 -1

transformers/data/processors/squad.py transformers/data/processors/squad.py +171 -109

No files found.
--- a/transformers/data/processors/glue.py
+++ b/transformers/data/processors/glue.py
@@ -133,7 +133,7 @@ def glue_convert_examples_to_features(examples, tokenizer,
    if is_tf_available() and is_tf_dataset:
        def gen():
            for ex in features:
-                yield  ({'input_ids': ex.input_ids,
+                yield ({'input_ids': ex.input_ids,
                         'attention_mask': ex.attention_mask,
                         'token_type_ids': ex.token_type_ids},
                        ex.label)

--- a/transformers/data/processors/squad.py
+++ b/transformers/data/processors/squad.py