Skip to content

Commit 545e705

Browse files
committed
chore: Allow numpy >= 2.0.0
A large e2e run with forced numpy 2.0+ suggests that the library is compatible with numpy 2.x series. Also, removed `numpy` dependency from requirements-dev.txt because it's already listed in requirements.txt. See: https://github.com/instructlab/training/actions/runs/14539292904 Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
1 parent 2a1e9b6 commit 545e705

3 files changed

Lines changed: 2 additions & 3 deletions

File tree

constraints-dev.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ nodeenv==1.9.1 # via pre-commit
109109
notebook==7.4.3 # via jupyter
110110
notebook-shim==0.2.4 # via jupyterlab, notebook
111111
numba==0.61.2 # via -r requirements.txt
112-
numpy==1.26.4 # via accelerate, bitsandbytes, contourpy, datasets, deepspeed, matplotlib, numba, pandas, peft, tensorboard, transformers, -r requirements-dev.txt, -r requirements.txt
112+
numpy==2.2.6 # via accelerate, bitsandbytes, contourpy, datasets, deepspeed, matplotlib, numba, pandas, peft, tensorboard, transformers, -r requirements-dev.txt, -r requirements.txt
113113
nvidia-cublas-cu12==12.4.5.8 # via nvidia-cudnn-cu12, nvidia-cusolver-cu12, torch
114114
nvidia-cuda-cupti-cu12==12.4.127 # via torch
115115
nvidia-cuda-nvrtc-cu12==12.4.127 # via torch

requirements-dev.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
isort
66
matplotlib
7-
numpy
87
pre-commit>=3.0.4
98
pylint>=2.16.2
109
pylint-pydantic

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ transformers>=4.45.2
77

88
datasets>=2.15.0
99
numba
10-
numpy>=1.26.4,<2.0.0
10+
numpy>=1.26.4
1111
rich
1212
instructlab-dolomite>=0.2.0
1313
trl>=0.9.4

0 commit comments

Comments
 (0)