Adding numpy restriction

triton-inference-server · Jun 5, 2024 · 97c32ce · 97c32ce
1 parent 7752666
commit 97c32ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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},
 )