Fix tokenizer test

keras-team · Mar 23, 2024 · 60bde7c · 60bde7c
1 parent 9c8c6f6
commit 60bde7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_nlp/models/electra/electra_tokenizer_test.py b/keras_nlp/models/electra/electra_tokenizer_test.py
@@ -59,7 +59,7 @@ def test_smallest_preset(self):
             cls=ElectraTokenizer,
             preset="electra_small_discriminator_en",
             input_data=["The quick brown fox."],
-            expected_output=[[1996, 4248, 2829, 4419, 1012]],
+            expected_output=[[100, 4248, 2829, 4419, 1012]],
         )
 
     @pytest.mark.extra_large