numpy 2.0 (#1748)

Lightning-AI · Feb 12, 2025 · 79c1c2b · 79c1c2b
1 parent 621dce7
commit 79c1c2b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/requirements/base.txt b/requirements/base.txt
@@ -1,7 +1,7 @@
 torch >=2.3.0
 looseversion ==1.3.0
 lightning-utilities >=0.7.0
-numpy >=1.23.0,<2  # not yet ready for numpy 2
+numpy
 networkx >= 3.3
 optree >=0.12.1
 opt_einsum >= 3.3.0

diff --git a/requirements/notebooks.txt b/requirements/notebooks.txt
@@ -1,5 +1,5 @@
 ipython[all] ~=8.30.0
-numpy >=1.23.0,<2  # not yet ready for numpy 2
+numpy
 liger-kernel == 0.4.0
 cuda-python
 litgpt == 0.5.1
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -7,10 +7,10 @@ pytest-xdist ==3.6.1
 pytest-random-order ==1.1.1
 pytest-timestamper ==0.0.10
 graphviz ==0.20.3
-fdm ==0.4.1
+fdm ==0.5.0
 expecttest ==0.3.0  # for test_ddp.py
 hypothesis ~=6.124.7  # for test_ddp.py
-numpy >=1.23.0,<2  # for test_ops.py; not yet ready for numpy 2
+numpy
 einops  # for test_einops.py
 litgpt==0.4.11  # for the model definition in tests and benchmarks
 absl-py # thunder/benchmarks/test_benchmark_litgpt.py