enable worker argument, but set to zero for tests

mllam · Nov 16, 2024 · f56a999 · f56a999
1 parent dfec1ec
commit f56a999
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/neural_lam/datastore/npyfilesmeps/compute_standardization_stats.py b/neural_lam/datastore/npyfilesmeps/compute_standardization_stats.py
@@ -193,7 +193,7 @@ def main(
         ds,
         batch_size,
         shuffle=False,
-        # num_workers=args.n_workers,
+        num_workers=n_workers,
         sampler=sampler,
     )
 

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -76,7 +76,7 @@ def download_meps_example_reduced_dataset():
             datastore_config_path=config_path,
             batch_size=8,
             step_length=3,
-            n_workers=1,
+            n_workers=0,
             distributed=False,
         )