Skip to content

Commit

Permalink
move the version to setup.py
Browse files Browse the repository at this point in the history
Signed-off-by: Andrey Stotskiy <[email protected]>
  • Loading branch information
RuRo committed Oct 26, 2021
1 parent b5b9be2 commit 38e7cd6
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 3 deletions.
8 changes: 7 additions & 1 deletion onnx_tensorrt/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,12 @@

from __future__ import absolute_import

import pkg_resources

from . import backend

__version__ = "8.2.0"
try:
# The version is defined in setup.py
__version__ = pkg_resources.require("onnx_tensorrt")[0].version
except pkg_resources.DistributionNotFound:
__version__ = None
3 changes: 1 addition & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

import os
import sys
import onnx_tensorrt
from setuptools import setup, find_packages

def no_publish():
Expand All @@ -22,7 +21,7 @@ def main():
no_publish()
setup(
name="onnx_tensorrt",
version=onnx_tensorrt.__version__,
version="8.2.0",
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",
Expand Down

0 comments on commit 38e7cd6

Please sign in to comment.