fix

2024-05-31 11:56:28 +00:00 · 2024-05-31 11:56:28 +00:00 · 9cf2856cea
parent 440cc84131
commit 9cf2856cea
1 changed files with 1 additions and 1 deletions
--- a/tests/test_policies.py
+++ b/tests/test_policies.py
@ -140,7 +140,7 @@ def test_policy(env_name, policy_name, extra_overrides):

    dataloader = torch.utils.data.DataLoader(
        dataset,
-        num_workers=4,
+        num_workers=0,
        batch_size=2,
        shuffle=True,
        pin_memory=DEVICE != "cpu",