fix issue with python 3.12

Signed-off-by: Xavier Dupre <xadupre@microsoft.com>
xadupre · May 22, 2024 · 05f76b9 · 05f76b9
1 parent ab14529
commit 05f76b9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/onnxmltools/proto/__init__.py b/onnxmltools/proto/__init__.py
@@ -15,13 +15,9 @@
 
 
 def _check_onnx_version():
-    import pkg_resources
+    from onnx import __version__
 
-    min_required_version = pkg_resources.parse_version("1.0.1")
-    current_version = pkg_resources.get_distribution("onnx").parsed_version
-    assert (
-        current_version >= min_required_version
-    ), "ONNXMLTools requires ONNX version 1.0.1 or a newer one"
+    return not __version__.startswith("0.")
 
 
 _check_onnx_version()