Skip to content

Commit 676c266

Browse files
committed
fix missing provider
Signed-off-by: xadupre <[email protected]>
1 parent 7130daf commit 676c266

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

tests/test_issues_2024.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,9 @@ def test_issue_1129_lr(self):
320320
model, x.values, options={"zipmap": False}
321321
)
322322
# Take predictions and probabilities with ONNX
323-
sess = InferenceSession(onnx_model.SerializeToString())
323+
sess = InferenceSession(
324+
onnx_model.SerializeToString(), providers=["CPUExecutionProvider"]
325+
)
324326
onnx_prediction = sess.run(None, {"X": x_test.to_numpy()})
325327
assert_almost_equal(sklearn_probs, onnx_prediction[1], decimal=decimal)
326328
assert_almost_equal(sklearn_preds, onnx_prediction[0])

tests/test_sklearn_pipeline_concat_tfidf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -379,8 +379,8 @@ def test_issue_712_svc_binary_empty(self):
379379
target_opset=TARGET_OPSET,
380380
options={CountVectorizer: {"keep_empty_string": True}},
381381
)
382-
with open("debug.onnx", "wb") as f:
383-
f.write(onx.SerializeToString())
382+
# with open("debug.onnx", "wb") as f:
383+
# f.write(onx.SerializeToString())
384384
sess = InferenceSession(
385385
onx.SerializeToString(), providers=["CPUExecutionProvider"]
386386
)

0 commit comments

Comments
 (0)