Add dev dependency for numpy<2 (#113)

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 17, 2024 · be0403c · be0403c
1 parent 222e5a2
commit be0403c
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]