Added support for 3.10 with ORT 1.22 and deprecation warning for the future

Signed-off-by: gcunhase <4861122+gcunhase@users.noreply.github.com>

Author: gcunhase

modelopt/onnx/__init__.py

@@ -16,6 +16,7 @@
 """Model optimization subpackage for onnx."""
 
 import sys
+import warnings
 
 MIN_PYTHON_VERSION = (3, 10)
 
@@ -26,6 +27,12 @@
     raise ImportError(f"{e}\nPlease install optional ``[onnx]`` dependencies.")
 
 
+if sys.version_info < (3, 11):
+    warnings.warn(
+        "`modelopt.onnx` package will drop python<3.11 support in a future release",
+        DeprecationWarning,
+    )
+
 # Check the current Python version
 if sys.version_info < MIN_PYTHON_VERSION:
     logger.warning(

setup.py

@@ -48,8 +48,13 @@
         "onnx-graphsurgeon",
         "onnx~=1.19.0",
         "onnxconverter-common~=1.16.0",
-        "onnxruntime~=1.24.2 ; platform_machine == 'aarch64' or platform_system == 'Darwin'",
-        "onnxruntime-gpu~=1.24.2 ; platform_machine != 'aarch64' and platform_system != 'Darwin' and platform_system != 'Windows'",  # noqa: E501
+        # ORT with Python > 3.10
+        "onnxruntime~=1.24.2 ; python_version > '3.10' and (platform_machine == 'aarch64' or platform_system == 'Darwin')",  # noqa: E501
+        "onnxruntime-gpu~=1.24.2 ; python_version > '3.10' and platform_machine != 'aarch64' and platform_system != 'Darwin' and platform_system != 'Windows'",  # noqa: E501
+        # ORT with Python <= 3.10
+        "onnxruntime~=1.22.0 ; python_version <= '3.10' and (platform_machine == 'aarch64' or platform_system == 'Darwin')",  # noqa: E501
+        "onnxruntime-gpu~=1.22.0 ; python_version <= '3.10' and platform_machine != 'aarch64' and platform_system != 'Darwin' and platform_system != 'Windows'",  # noqa: E501
+        # ORT for Windows
         "onnxruntime-gpu==1.22.0; platform_system == 'Windows'",
         "onnxscript",  # For autocast opset conversion and test_onnx_dynamo_export unit test
         "onnxslim>=0.1.76",