diff --git a/tests/test_issues_2024.py b/tests/test_issues_2024.py index 51ea09472..1cf6fa94d 100644 --- a/tests/test_issues_2024.py +++ b/tests/test_issues_2024.py @@ -320,7 +320,9 @@ def test_issue_1129_lr(self): model, x.values, options={"zipmap": False} ) # Take predictions and probabilities with ONNX - sess = InferenceSession(onnx_model.SerializeToString()) + sess = InferenceSession( + onnx_model.SerializeToString(), providers=["CPUExecutionProvider"] + ) onnx_prediction = sess.run(None, {"X": x_test.to_numpy()}) assert_almost_equal(sklearn_probs, onnx_prediction[1], decimal=decimal) assert_almost_equal(sklearn_preds, onnx_prediction[0]) diff --git a/tests/test_sklearn_pipeline_concat_tfidf.py b/tests/test_sklearn_pipeline_concat_tfidf.py index 03b88f11e..c9c7b96f5 100644 --- a/tests/test_sklearn_pipeline_concat_tfidf.py +++ b/tests/test_sklearn_pipeline_concat_tfidf.py @@ -379,8 +379,8 @@ def test_issue_712_svc_binary_empty(self): target_opset=TARGET_OPSET, options={CountVectorizer: {"keep_empty_string": True}}, ) - with open("debug.onnx", "wb") as f: - f.write(onx.SerializeToString()) + # with open("debug.onnx", "wb") as f: + # f.write(onx.SerializeToString()) sess = InferenceSession( onx.SerializeToString(), providers=["CPUExecutionProvider"] )