diff --git a/tests/test_training.py b/tests/test_training.py index 7ad28e4d..f02bef4e 100644 --- a/tests/test_training.py +++ b/tests/test_training.py @@ -63,7 +63,7 @@ def test_training(datastore_name): ar_steps_eval=5, standardize=True, batch_size=2, - num_workers=1, + num_workers=0, num_past_forcing_steps=1, num_future_forcing_steps=1, )