Update tests/openvino/test_modeling.py

huggingface · Oct 28, 2024 · 7f2f404 · 7f2f404
1 parent 9fab315
commit 7f2f404
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -674,6 +674,7 @@ def test_default_token_type_ids(self):
         tokenizer = AutoTokenizer.from_pretrained(model_id)
         tokens = tokenizer("this is a simple input", return_tensors="np")
         self.assertTrue("token_type_ids" in model.input_names)
+        token_type_ids = tokens.pop("token_type_ids")
         outs = model(token_type_ids=token_type_ids, **tokens)
         outs_without_token_type_ids = model(**tokens)
         self.assertTrue(np.allclose(outs.logits, outs_without_token_type_ids.logits))