Bound numpy version below 2.2.0 (#849)

mosaicml · Dec 9, 2024 · 67d0cbf · 67d0cbf
1 parent 93af28a
commit 67d0cbf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -46,6 +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',
     'paramiko>=2.11.0,<4',
     'python-snappy>=0.6.1,<1',
     'torch>=1.10,<3',