commit 9cd45595da12573d19797368fec767545509bb97
author Jonas Ohlsson <jonas.ohlsson@arm.com> Tue Aug 24 11:11:55 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Tue Aug 24 11:18:00 2021 +0000
tree 2b091e3c549da83a02c2505584a1f3b4469fd730
parent 3b139a478693e7461281dfa80cb0a5287c721c41

MLBEDSW-5108 Update setup.py to fix install failures

To avoid using Python 3.6 incompatible versions of NumPy (> 1.19.5),
an upper bound on version is added for NumPy in setup.py.

Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: I3929bd7dbea6866905665186af1c4b3ba43ccbd0

setup.py

diff --git a/setup.py b/setup.py
index d4bc009..2aa23ca 100644
--- a/setup.py
+++ b/setup.py
@@ -91,12 +91,12 @@
     python_requires="~=3.6",
     install_requires=[
         "flatbuffers==1.12.0",
-        "numpy>=1.16.6",
+        "numpy>=1.16.6,<=1.19.5",
         "numpy>=1.16.6,<1.19.4 ; platform_system=='Windows'",
         "lxml>=4.5.1",
     ],
     entry_points={"console_scripts": ["vela = ethosu.vela.vela:main"]},
     ext_modules=[mlw_module],
     cmdclass={"build_ext": BuildExtension},
-    setup_requires=["numpy>=1.16.6", "setuptools_scm"],
+    setup_requires=["numpy>=1.16.6,<=1.19.5", "setuptools_scm"],
 )