Unverified Commit 35677eca authored by Thiago Crepaldi's avatar Thiago Crepaldi Committed by GitHub
Browse files

Update InferenceSession call to latest format (#7982)


Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
parent cb512ea6
...@@ -124,7 +124,7 @@ jobs: ...@@ -124,7 +124,7 @@ jobs:
conda activate ci conda activate ci
echo '::group::Install ONNX' echo '::group::Install ONNX'
pip install --progress-bar=off onnx onnxruntime!=1.16.0 pip install --progress-bar=off onnx onnxruntime
echo '::endgroup::' echo '::endgroup::'
echo '::group::Install testing utilities' echo '::group::Install testing utilities'
......
...@@ -79,7 +79,7 @@ class TestONNXExporter: ...@@ -79,7 +79,7 @@ class TestONNXExporter:
inputs = list(map(to_numpy, inputs)) inputs = list(map(to_numpy, inputs))
outputs = list(map(to_numpy, outputs)) 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 # compute onnxruntime output prediction
ort_inputs = {ort_session.get_inputs()[i].name: inpt for i, inpt in enumerate(inputs)} ort_inputs = {ort_session.get_inputs()[i].name: inpt for i, inpt in enumerate(inputs)}
ort_outs = ort_session.run(None, ort_inputs) ort_outs = ort_session.run(None, ort_inputs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment