Fewer num_workers

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