From 05f76b940f6393ce1b3618330196073b2cc65beb Mon Sep 17 00:00:00 2001 From: Xavier Dupre Date: Wed, 22 May 2024 11:56:13 +0200 Subject: [PATCH] fix issue with python 3.12 Signed-off-by: Xavier Dupre --- onnxmltools/proto/__init__.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/onnxmltools/proto/__init__.py b/onnxmltools/proto/__init__.py index a6bd5faf..e84b6419 100644 --- 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()