Add dev dependency for numpy<2

The pytorch lightning version we're using for training is incompatible with numpy 2. We should update the training code to work with newer lightning, but in the meantime set an upper bound for numpy in the dev deps.
unitaryai · Aug 15, 2024 · 1026f5f · 1026f5f
1 parent 222e5a2
commit 1026f5f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,6 +33,7 @@ dev = [
     "scikit-learn >= 0.23.2",
     "tqdm",
     "pre-commit",
+    "numpy<2"
 ]
 
 [tool.black]