Skip to content

Commit 99b42ad

Browse files
committed
remove tcp port
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent b08222b commit 99b42ad

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

bionemo-recipes/recipes/llama3_native_te/tests/test_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -750,7 +750,7 @@ def test_cp_dataloader(tokenizer_path):
750750

751751
@requires_multi_gpu
752752
@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):
753+
def test_cp_dataloader_multi_gpu(recipe_path, dataset_path):
754754
"""Tests that the CP dataloader works correctly with multiple GPUs.
755755
756756
The `test_genomic_sequences.parquet` dataset is too small to even fill a single batch with the default context

0 commit comments

Comments
 (0)