Merge pull request #93 from artemisart/fix-test

fix tokenization_test
hsm207 · Nov 9, 2018 · 35deae7 · 35deae7
2 parents 3c67c1d + ff7d05a
commit 35deae7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tokenization_test.py b/tokenization_test.py
@@ -30,7 +30,7 @@ def test_full_tokenizer(self):
  "[UNK]", "[CLS]", "[SEP]", "want", "##want", "##ed", "wa", "un", "runn",
  "##ing", ","
  ]
- with tempfile.NamedTemporaryFile(delete=False) as vocab_writer:
+ with tempfile.NamedTemporaryFile(mode='w+', delete=False) as vocab_writer:
  vocab_writer.write("".join([x + "\n" for x in vocab_tokens]))
 
  vocab_file = vocab_writer.name