diff --git a/bindings/python/tests/bindings/test_models.py b/bindings/python/tests/bindings/test_models.py index 7b6746186..063698384 100644 --- a/bindings/python/tests/bindings/test_models.py +++ b/bindings/python/tests/bindings/test_models.py @@ -73,6 +73,7 @@ def test_dropout_zero(self): model = BPE(dropout=0.0) assert model.dropout == 0.0 + class TestWordPiece: def test_instantiate(self, bert_files): assert isinstance(WordPiece(), Model)