We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b08222b commit 99b42adCopy full SHA for 99b42ad
1 file changed
bionemo-recipes/recipes/llama3_native_te/tests/test_dataset.py
@@ -750,7 +750,7 @@ def test_cp_dataloader(tokenizer_path):
750
751
@requires_multi_gpu
752
@pytest.mark.parametrize("dataset_path", ["dlcm_sanity_dataset.parquet", "test_genomic_sequences.parquet"])
753
-def test_cp_dataloader_multi_gpu(recipe_path, dataset_path, unused_tcp_port):
+def test_cp_dataloader_multi_gpu(recipe_path, dataset_path):
754
"""Tests that the CP dataloader works correctly with multiple GPUs.
755
756
The `test_genomic_sequences.parquet` dataset is too small to even fill a single batch with the default context
0 commit comments