diff --git a/tests/models/test_onnx.py b/tests/models/test_onnx.py
index 95732cfd8f..32d46607b5 100644
--- a/tests/models/test_onnx.py
+++ b/tests/models/test_onnx.py
@@ -85,7 +85,7 @@ def test_onnx_export(tie_word_embeddings: bool, tmp_path: pathlib.Path):
     ort_session = ort.InferenceSession(str(tmp_path / 'mpt.onnx'))
 
     for key, value in sample_input.items():
-        sample_input[key] = value.cpu().numpy()
+        sample_input[key] = value.cpu().numpy()  # pyright: ignore
 
     loaded_model_out = ort_session.run(None, sample_input)