diff --git a/setup.py b/setup.py
index 92999d676..cc13d8374 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@
     'Brotli>=1.0.9',
     'google-cloud-storage>=2.9.0,<2.11.0',
     'matplotlib>=3.5.2,<4',
-    'numpy>=1.21.5,<2.2.0',
+    'numpy>=1.21.5,<2.3.0',
     'paramiko>=2.11.0,<4',
     'python-snappy>=0.6.1,<1',
     'torch>=1.10,<3',