diff --git a/onnx_tensorrt/__init__.py b/onnx_tensorrt/__init__.py index 5f3e42c..b70cfe7 100644 --- a/onnx_tensorrt/__init__.py +++ b/onnx_tensorrt/__init__.py @@ -4,4 +4,4 @@ from . import backend -__version__ = "10.13.0" +from .version import __version__ diff --git a/onnx_tensorrt/version.py b/onnx_tensorrt/version.py new file mode 100644 index 0000000..45225b2 --- /dev/null +++ b/onnx_tensorrt/version.py @@ -0,0 +1 @@ +__version__ = "10.13.0" \ No newline at end of file diff --git a/setup.py b/setup.py index ad101a9..0f2a767 100644 --- a/setup.py +++ b/setup.py @@ -1,8 +1,7 @@ # SPDX-License-Identifier: Apache-2.0 -import os +import re import sys -import onnx_tensorrt from setuptools import setup, find_packages def no_publish(): @@ -11,6 +10,14 @@ def no_publish(): if cmd in sys.argv: raise RuntimeError("Command \"{}\" blacklisted".format(cmd)) +def read_version(filepath = 'onnx_tensorrt/version.py'): + try: + with open(filepath) as f: + version_text = f.read() + version, = re.findall('__version__ = "(.*)"', version_text) + return version + except Exception as error: + raise RuntimeError(f"Failed to read version from {filepath}: {error}") from error REQUIRED_PACKAGES = [ "pycuda", @@ -22,7 +29,7 @@ def main(): no_publish() setup( name="onnx_tensorrt", - version=onnx_tensorrt.__version__, + version=read_version('onnx_tensorrt/version.py'), description="ONNX-TensorRT - TensorRT backend for running ONNX models", long_description=open("README.md", "r", encoding="utf-8").read(), url="https://github.com/onnx/onnx-tensorrt",