From 60bde7cfaa517c6d26e07ffbbb13724a8e88df6b Mon Sep 17 00:00:00 2001 From: Matt Watson Date: Fri, 22 Mar 2024 17:26:56 -0700 Subject: [PATCH] Fix tokenizer test --- keras_nlp/models/electra/electra_tokenizer_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_nlp/models/electra/electra_tokenizer_test.py b/keras_nlp/models/electra/electra_tokenizer_test.py index 84ef074ed1..e09f33c0b8 100644 --- 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