From 673cf11bab7425429dc4c3a188e54c2976ee44f6 Mon Sep 17 00:00:00 2001 From: William Fondrie Date: Tue, 16 Apr 2024 16:12:47 -0700 Subject: [PATCH] Fewer num_workers --- tests/unit_tests/test_data/test_loaders.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit_tests/test_data/test_loaders.py b/tests/unit_tests/test_data/test_loaders.py index d802eaf..2a1c84a 100644 --- a/tests/unit_tests/test_data/test_loaders.py +++ b/tests/unit_tests/test_data/test_loaders.py @@ -32,7 +32,7 @@ def test_streaming_spectrum_loader(mgf_small, tmp_path): stream_batch = next(iter(loader)) dset = SpectrumDataset(mgf_small, batch_size=2, path=tmp_path / "test") - loader = DataLoader(dset, num_workers=1) + loader = DataLoader(dset) map_batch = next(iter(loader)) assert_dicts_equal(stream_batch, map_batch) @@ -68,7 +68,7 @@ def test_analyte_loader(): charges = torch.tensor([5, 3, 1]) tokenizer = PeptideTokenizer() dset = AnalyteDataset(tokenizer, seqs, charges) - loader = dset.loader(batch_size=2, num_workers=0) + loader = dset.loader(batch_size=2) batch = next(iter(loader)) assert len(batch) == 2 @@ -80,7 +80,7 @@ def test_analyte_loader(): args = (torch.tensor([1, 2, 3]), torch.tensor([[1, 1], [2, 2], [3, 3]])) dset = AnalyteDataset(tokenizer, seqs, charges, *args) - loader = dset.loader(batch_size=2, num_workers=0) + loader = dset.loader(batch_size=2) batch = next(iter(loader)) assert len(batch) == 4