Relax ORT requirements (#147)

* Relax ORT version requirements * Ensure conda env and base env match
onnx · Apr 1, 2024 · 7c13aa6 · 7c13aa6
1 parent b814d0b
commit 7c13aa6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -36,7 +36,7 @@
         "scikit-learn==1.1.1",
         "xgboost==1.6.1",
         "lightgbm==3.3.5",
-        "onnxruntime >=1.10.1, <1.16.0",
+        "onnxruntime >=1.10.1",
         "torch>=1.12.1",
         "protobuf>=3.17.3,<3.21",
         "pyyaml>=5.4",

diff --git a/src/turnkeyml/run/onnxrt/execute.py b/src/turnkeyml/run/onnxrt/execute.py
@@ -9,9 +9,11 @@
 import json
 from statistics import mean
 import platform
+import onnxruntime
 import turnkeyml.run.plugin_helpers as plugin_helpers
 
-ORT_VERSION = "1.15.1"
+# Use the same ORT version of the base environment in the new conda env
+ORT_VERSION = onnxruntime.__version__
 
 BATCHSIZE = 1
 

diff --git a/src/turnkeyml/version.py b/src/turnkeyml/version.py
@@ -1 +1 @@
-__version__ = "2.0.1"
+__version__ = "2.0.2"