Skip to content

Commit 2ab3700

Browse files
committed
fix: add pytest to requirements.txt for recipes missing it
The pytorch:26.04-py3 base image no longer ships pytest pre-installed. This adds pytest to all requirements.txt files that were relying on the base image providing it, fixing 'pytest: command not found' CI failures. Signed-off-by: svc-bionemo <267129667+svc-bionemo@users.noreply.github.com>
1 parent f0d9406 commit 2ab3700

12 files changed

Lines changed: 12 additions & 0 deletions

File tree

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
torch
22
transformer_engine[pytorch]
33
transformers
4+
pytest

bionemo-recipes/models/esm2/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,3 +9,4 @@ torch
99
torchao!=0.14.0
1010
transformer_engine[pytorch]
1111
transformers
12+
pytest

bionemo-recipes/models/llama3/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@ torch
33
torchao!=0.14.0
44
transformer_engine[pytorch]
55
transformers
6+
pytest

bionemo-recipes/models/mixtral/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@ torch
33
torchao!=0.14.0
44
transformer_engine[pytorch]
55
transformers
6+
pytest

bionemo-recipes/models/qwen/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@ torch
33
torchao!=0.14.0
44
transformer_engine[pytorch]
55
transformers
6+
pytest

bionemo-recipes/recipes/esm2_accelerate_te/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,4 @@ torchmetrics
77
transformer_engine[pytorch]
88
transformers
99
wandb
10+
pytest

bionemo-recipes/recipes/esm2_peft_te/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,4 @@ tqdm
88
transformer_engine[pytorch]
99
transformers
1010
wandb
11+
pytest

bionemo-recipes/recipes/fp8_analysis/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,4 @@ matplotlib==3.10.1
22
numpy==1.26.4
33
pandas==2.2.3
44
seaborn==0.13.2
5+
pytest

bionemo-recipes/recipes/geneformer_native_te_mfsdp_fp8/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,4 @@ tqdm
88
transformer_engine
99
transformers<5.0
1010
wandb
11+
pytest

bionemo-recipes/recipes/llama3_native_te/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,4 @@ transformers
1010
wandb
1111
zstandard
1212
nvdlfw_inspect @ git+https://github.com/NVIDIA/nvidia-dlfw-inspect
13+
pytest

0 commit comments

Comments
 (0)