Mark geneformer test_pretrain_cli as slow

This is one remaining outlier in our current test suite that takes a long time to run (~3 minutes by itself). https://app.codecov.io/gh/NVIDIA/bionemo-framework/tests/main With #634 we'll run these tests anyways during the merge queue step, but marking this as slow should speed up the coverage runs in PRs and post-merge Signed-off-by: Peter St. John <[email protected]>
NVIDIA · Jan 23, 2025 · d16b10d · d16b10d
1 parent 17c6b20
commit d16b10d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/scripts/test_train_geneformer.py b/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/scripts/test_train_geneformer.py
@@ -181,6 +181,7 @@ def test_throws_tok_not_in_vocab_error(tmpdir, data_path):
     assert "not in the tokenizer vocab." in str(error_info.value)
 
 
+@pytest.mark.slow
 def test_pretrain_cli(tmpdir, data_path):
     result_dir = Path(tmpdir.mkdir("results"))
     open_port = find_free_network_port()