diff --git a/python/setup.py b/python/setup.py index 61a8590c0..82111551e 100755 --- a/python/setup.py +++ b/python/setup.py @@ -108,6 +108,6 @@ def get_tag(self): zip_safe=False, cmdclass={"bdist_wheel": bdist_wheel}, data_files=data_files, - install_requires=["numpy"], + install_requires=["numpy<2"], extras_require={"GPU": gpu_extras, "test": test_extras, "all": all_extras}, )