diff --git a/tests/test_datasets.py b/tests/test_datasets.py index ea72f682..236a7032 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -115,7 +115,7 @@ def test_compute_stats_on_xarm(): # get all frames from the dataset in the same dtype and range as during compute_stats dataloader = torch.utils.data.DataLoader( dataset, - num_workers=8, + num_workers=1, batch_size=len(dataset), shuffle=False, )