address comments

keras-team · Oct 19, 2022 · 205efa0 · 205efa0
1 parent 5ba5534
commit 205efa0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/keras_nlp/tokenizers/byte_pair_tokenizer.py b/keras_nlp/tokenizers/byte_pair_tokenizer.py
@@ -234,7 +234,7 @@ def __init__(
         self,
         vocabulary,
         merges,
-        sequence_length: int = None,
+        sequence_length=None,
         **kwargs,
     ) -> None:
         # Check dtype and provide a default.

diff --git a/keras_nlp/tokenizers/byte_pair_tokenizer_test.py b/keras_nlp/tokenizers/byte_pair_tokenizer_test.py
@@ -14,6 +14,7 @@
 
 import os
 
+import pytest
 import tensorflow as tf
 from absl.testing import parameterized
 from tensorflow import keras
@@ -30,7 +31,7 @@
 )
 
 
-# @pytest.mark.slow
+@pytest.mark.slow
 class BytePairTokenizerTest(tf.test.TestCase, parameterized.TestCase):
     def setUp(self):
         super().setUp()