yo

mosaicml · Aug 30, 2024 · b4d51a5 · b4d51a5
1 parent ca26c51
commit b4d51a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)