From 499ab58ebf059e399eccf118e4012825f22b88d1 Mon Sep 17 00:00:00 2001
From: Lester Solbakken <lesters@oath.com>
Date: Thu, 21 Sep 2023 12:19:32 +0200
Subject: [PATCH] Specify ONNX execution provider

---
 learntorank/test_ml.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/learntorank/test_ml.py b/learntorank/test_ml.py
index 61557f0..977362c 100644
--- a/learntorank/test_ml.py
+++ b/learntorank/test_ml.py
@@ -111,7 +111,7 @@ def _predict_with_onnx(onnx_file_path, model_inputs):
         os.environ[
             "KMP_DUPLICATE_LIB_OK"
         ] = "True"  # required to run on mac https://stackoverflow.com/a/53014308
-        m = InferenceSession(onnx_file_path)
+        m = InferenceSession(onnx_file_path, providers=['CPUExecutionProvider'])
         (out,) = m.run(input_feed=model_inputs, output_names=["output_0"])
         return out