fix missing provider

onnx · Oct 3, 2024 · d111679 · d111679
1 parent 9970498
commit d111679
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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"]
         )