Skip to content

Commit d7d066f

Browse files
committed
Adapt Python bindings
1 parent 47c8472 commit d7d066f

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

setup.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,10 +102,18 @@ def build_extension(self, ext):
102102
long_description = f.read()
103103

104104
# read version
105-
f = open("version", "r")
105+
f = open("version.txt", "r")
106106
pbd_version = f.readline().strip()
107107
f.close()
108108

109+
max_numpy_version = "1.19.3"
110+
install_requires = [f"numpy<={max_numpy_version}"]
111+
if platform.machine() == "arm64":
112+
if platform.system() == "Darwin":
113+
install_requires = ["numpy>=1.21.0"]
114+
else:
115+
install_requires = [f"numpy>=1.19.0,<={max_numpy_version}"]
116+
109117
setup(
110118
name=name,
111119
version=pbd_version,
@@ -121,5 +129,5 @@ def build_extension(self, ext):
121129
cmdclass=dict(build_ext=CMakeBuild),
122130
packages=find_packages(),
123131
zip_safe=False,
124-
install_requires=['numpy<=1.19.3']
132+
install_requires=install_requires
125133
)

0 commit comments

Comments
 (0)