fix batch distribution for other worker types

openforcefield · Feb 9, 2024 · 6bd3c95 · 6bd3c95
1 parent fd2d78f
commit 6bd3c95
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/openff/nagl/utils/_parallelization.py b/openff/nagl/utils/_parallelization.py
@@ -118,9 +118,11 @@ def batch_distributed(
         f"Setting n_workers={n_workers} for {n_batches} batches"
     )
 
-    env_extra = dask.config.get("jobqueue.lsf.job-script-prologue", default=[])
-    if not env_extra:
-        env_extra = []
+    env_extra = []
+    if worker_type != "local":
+        env_extra.extend(
+            dask.config.get("jobqueue.job-script-prologue", default=[])
+        )
     env_extra.append(f"{package_manager} activate {conda_environment}")