Apply suggestions from code review

Lightning-AI · Aug 14, 2024 · da8d3e4 · da8d3e4
1 parent 5586fe4
commit da8d3e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,7 @@ classifiers = [
 [project.optional-dependencies]
 all = [
     "tqdm",  # convert_checkpoint.py
-    "numpy",  # train.py dataset memmap
+    "numpy <2.0",  # train.py dataset memmap
     "jsonargparse[signatures]",  # generate.py, convert_checkpoint.py CLI
     "datasets",  # evaluate.py
     "zstandard",  # prepare_redpajama.py"