Update InferenceSession call to latest format (#7982)

Co-authored-by: Philip Meier <github.pmeier@posteo.de>
pytorch · Sep 22, 2023 · 35677ec · 35677ec
1 parent cb512ea
commit 35677ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -124,7 +124,7 @@ jobs:
         conda activate ci
         
         echo '::group::Install ONNX'
-        pip install --progress-bar=off onnx onnxruntime!=1.16.0
+        pip install --progress-bar=off onnx onnxruntime
         echo '::endgroup::'
         
         echo '::group::Install testing utilities'

diff --git a/test/test_onnx.py b/test/test_onnx.py
@@ -79,7 +79,7 @@ def to_numpy(tensor):
         inputs = list(map(to_numpy, inputs))
         outputs = list(map(to_numpy, outputs))
 
-        ort_session = onnxruntime.InferenceSession(onnx_io.getvalue())
+        ort_session = onnxruntime.InferenceSession(onnx_io.getvalue(), providers=onnxruntime.get_available_providers())
         # compute onnxruntime output prediction
         ort_inputs = {ort_session.get_inputs()[i].name: inpt for i, inpt in enumerate(inputs)}
         ort_outs = ort_session.run(None, ort_inputs)