From c2c3102aba365b1890c358ccce1d479bea2ec1d1 Mon Sep 17 00:00:00 2001 From: Colin Jermain Date: Sun, 12 Apr 2020 21:44:21 -0400 Subject: [PATCH] Tying install_requires to requirements.txt --- requirements.txt | 5 +++-- setup.py | 8 ++++---- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/requirements.txt b/requirements.txt index 1cf243d912..882788d812 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ -six -numpy +numpy >= 1.18.0 +onnx >= 1.2.3 protobuf +six diff --git a/setup.py b/setup.py index 34d77b699f..87180cff61 100644 --- a/setup.py +++ b/setup.py @@ -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',