Read SP mododel file in bytes mode in tests

tensorflow · Oct 9, 2020 · 8621b71 · 8621b71
1 parent 8539524
commit 8621b71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_text/python/ops/sentencepiece_tokenizer_test.py b/tensorflow_text/python/ops/sentencepiece_tokenizer_test.py
@@ -98,7 +98,7 @@ def setUp(self):
     super(SentencepieceTokenizerOpTest, self).setUp()
     sentencepiece_model_file = (
         'tensorflow_text/python/ops/test_data/test_oss_model.model')
-    self.model = gfile.GFile(sentencepiece_model_file, 'r').read()
+    self.model = gfile.GFile(sentencepiece_model_file, 'rb').read()
 
   def testGetVocabSize(self):
     sp = SentencepieceTokenizer(self.model)