From 97c32ceaee91bb98d35d52c19862d3d3ca155cae Mon Sep 17 00:00:00 2001 From: Krishnan Prashanth Date: Wed, 5 Jun 2024 15:20:50 -0700 Subject: [PATCH] Adding numpy restriction --- python/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}, )