From 18a70456bafb0701e5ff2ecf8e7f3668f8af7688 Mon Sep 17 00:00:00 2001
From: Lily Wang <lily@mdanalysis.org>
Date: Fri, 9 Feb 2024 15:49:35 +1100
Subject: [PATCH] update changelog

---
 CHANGELOG.md                          | 1 +
 openff/nagl/utils/_parallelization.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 5f87ef85..cbd77236 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -25,6 +25,7 @@ The rules for this file:
 
 ### Fixed
 <!-- Bug fixes -->
+- Fixed batch distribution hardcoding (Issue #80, PR #82)
 
 ### Changed
 <!-- Changes in existing functionality -->
diff --git a/openff/nagl/utils/_parallelization.py b/openff/nagl/utils/_parallelization.py
index 5fc6863c..d74e29a0 100644
--- a/openff/nagl/utils/_parallelization.py
+++ b/openff/nagl/utils/_parallelization.py
@@ -121,7 +121,7 @@ def batch_distributed(
     env_extra = []
     if worker_type != "local":
         env_extra.extend(
-            dask.config.get("jobqueue.job-script-prologue", default=[])
+            dask.config.get(f"jobqueue.{worker_type}.job-script-prologue", default=[])
         )
     env_extra.append(f"{package_manager} activate {conda_environment}")