zhutian
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import os
|
|
import setuptools
|
|
import subprocess
|
|
from textwrap import dedent
|
|
|
|
TOP_DIR = os.path.realpath(os.path.dirname(__file__))
|
|
SRC_DIR = os.path.join(TOP_DIR, 'onnx_tf')
|
|
|
|
with open(os.path.join(TOP_DIR, 'VERSION_NUMBER')) as version_file:
|
|
version = version_file.read().strip()
|
|
|
|
if os.getenv('TRAVIS'):
|
|
# On travis, we install from source, therefore no need to specify version.
|
|
onnx_dep = "onnx"
|
|
else:
|
|
# For user, we install the onnx release known to work with our release.
|
|
with open(os.path.join(TOP_DIR, 'ONNX_VERSION_NUMBER')) as onnx_version_file:
|
|
onnx_version = onnx_version_file.read().strip()
|
|
onnx_dep = "onnx>=" + onnx_version
|
|
|
|
try:
|
|
git_version = subprocess.check_output(['git', 'rev-parse', 'HEAD'],
|
|
cwd=TOP_DIR).decode('ascii').strip()
|
|
except (OSError, subprocess.CalledProcessError):
|
|
git_version = None
|
|
|
|
with open(os.path.join(SRC_DIR, 'version.py'), 'w') as f:
|
|
f.write(dedent('''\
|
|
# This file is generated by setup.py. DO NOT EDIT!
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
version = '{}'
|
|
git_version = '{}'
|
|
'''.format(version, git_version)))
|
|
|
|
setuptools.setup(
|
|
name='onnx-tf',
|
|
version=version,
|
|
description=
|
|
'Tensorflow backend for ONNX (Open Neural Network Exchange).',
|
|
install_requires=[onnx_dep, "PyYAML"],
|
|
entry_points={
|
|
"console_scripts": [
|
|
"onnx-tf=onnx_tf.cli:main",
|
|
],
|
|
},
|
|
url='https://github.com/onnx/onnx-tensorflow/',
|
|
author='Arpith Jacob, Tian Jin, Gheorghe-Teodor Bercea, Wenhao Hu',
|
|
author_email='tian.jin1@ibm.com',
|
|
license='Apache License 2.0',
|
|
packages=setuptools.find_packages(),
|
|
zip_safe=False,
|
|
classifiers=[
|
|
"Programming Language :: Python :: 2",
|
|
"Programming Language :: Python :: 3"
|
|
]
|
|
)
|