Fix numpy version for Python 3.6 (#3156)

* Fix numpy version for Python 3.6 * Limit version on Windows to <1.19.4

Fix numpy version for Python 3.6 (#3156)
* Fix numpy version for Python 3.6 * Limit version on Windows to <1.19.4
e12f6113 · Yuge Zhang · GitHub · 20eecf0b · e12f6113
Unverified Commit e12f6113 authored Dec 05, 2020 by Yuge Zhang Committed by GitHub Dec 05, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

setup.py setup.py +4 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -77,6 +77,10 @@ dependencies = [
    'prettytable'
 ]

+if sys.platform == 'win32':
+    dependencies[dependencies.index('numpy')] = 'numpy<1.19.4'
+elif sys.version_info < (3, 7):
+    dependencies[dependencies.index('numpy')] = 'numpy<1.20'

 release = os.environ.get('NNI_RELEASE')