Tying install_requires to requirements.txt

This commit is contained in:
Colin Jermain 2020-04-12 21:44:21 -04:00 committed by Changming Sun
parent 66a79d2c9f
commit c2c3102aba
2 changed files with 7 additions and 6 deletions

View file

@ -1,3 +1,4 @@
six
numpy
numpy >= 1.18.0
onnx >= 1.2.3
protobuf
six

View file

@ -201,6 +201,9 @@ if bdist_wheel is not None :
cmd_classes['bdist_wheel'] = bdist_wheel
cmd_classes['build_ext'] = build_ext
with open('requirements.txt') as f:
install_requires = f.read().splitlines()
# Setup
setup(
name=package_name,
@ -222,10 +225,7 @@ setup(
'onnxruntime': data + examples + extra,
},
py_modules=python_modules_list,
install_requires=[
'onnx>=1.2.3',
'numpy>=1.18.0'
],
install_requires=install_requires,
entry_points= {
'console_scripts': [
'onnxruntime_test = onnxruntime.tools.onnxruntime_test:main',