Unverified Commit 0b7225e9 authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Add `ort_nightly_directml` to the `onnxruntime` candidates (#1458)

* Add `ort_nightly_directml` to the `onnxruntime` candidates

* style
parent db7b7bd9
......@@ -157,7 +157,13 @@ except importlib_metadata.PackageNotFoundError:
_onnxruntime_version = "N/A"
_onnx_available = importlib.util.find_spec("onnxruntime") is not None
if _onnx_available:
candidates = ("onnxruntime", "onnxruntime-gpu", "onnxruntime-directml", "onnxruntime-openvino")
candidates = (
"onnxruntime",
"onnxruntime-gpu",
"onnxruntime-directml",
"onnxruntime-openvino",
"ort_nightly_directml",
)
_onnxruntime_version = None
# For the metadata, we have to look for both onnxruntime and onnxruntime-gpu
for pkg in candidates:
......
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