diff --git a/onnx_tensorrt/__init__.py b/onnx_tensorrt/__init__.py index 5f3e42c..4dd3f3f 100644 --- a/onnx_tensorrt/__init__.py +++ b/onnx_tensorrt/__init__.py @@ -1,7 +1,8 @@ # SPDX-License-Identifier: Apache-2.0 - -from __future__ import absolute_import - +from importlib.metadata import version as _v, PackageNotFoundError # Py3.8+ from . import backend -__version__ = "10.13.0" +try: + __version__ = _v("onnx_tensorrt") +except PackageNotFoundError: + __version__ = "0+unknown" diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..6142215 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,20 @@ +[build-system] +requires = ["setuptools>=69", "wheel"] +build-backend = "setuptools.build_meta" + +[project] +name = "onnx_tensorrt" +version = "10.13.0" +description = "ONNX-TensorRT - TensorRT backend for running ONNX models" +readme = { file = "README.md", content-type = "text/markdown" } +license = { text = "Apache-2.0" } +authors = [{ name = "NVIDIA", email = "svc_tensorrt@nvidia.com" }] +classifiers = [ + "Intended Audience :: Developers", + "Programming Language :: Python :: 3", +] +dependencies = ["pycuda", "numpy", "onnx"] + +[tool.setuptools.packages.find] +where = ["."] +include = ["onnx_tensorrt*"]