tests fix

keras-team · Aug 10, 2023 · e03e9bb · e03e9bb
1 parent 983d93d
commit e03e9bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_nlp/models/xlnet/xlnet_tokenizer_test.py b/keras_nlp/models/xlnet/xlnet_tokenizer_test.py
@@ -50,9 +50,9 @@ def setUp(self):
         self.tokenizer = XLNetTokenizer(proto=self.proto)
 
     def test_tokenize(self):
-        input_data = "the quick brown fox"
+        input_data = ["the quick brown fox"]
         output = self.tokenizer(input_data)
-        self.assertAllEqual(output, [7, 12, 8, 10, 6, 5])
+        self.assertAllEqual(output, [[7, 12, 8, 10, 6, 5]])
 
     def test_tokenize_batch(self):
         input_data = ["the quick brown fox", "the earth is round"]