Merge pull request #93 from artemisart/fix-test

fix tokenization_test

Merge pull request #93 from artemisart/fix-test
fix tokenization_test
35deae77 · Jacob Devlin · GitHub · 3c67c1de · ff7d05af · 35deae77
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tokenization_test.py tokenization_test.py +1 -1

未找到文件。
--- a/tokenization_test.py
+++ b/tokenization_test.py
@@ -30,7 +30,7 @@ class TokenizationTest(tf.test.TestCase):
        "[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