wip

esm-tools · Jan 9, 2025 · 09f4320 · 09f4320
1 parent 2b6a375
commit 09f4320
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/pymorize/cmorizer.py b/src/pymorize/cmorizer.py
@@ -244,8 +244,10 @@ def _post_init_populate_rules_with_data_request_variables(self):
             else:
                 rule_for_var.data_request_variables.append(drv)
         # FIXME: This needs a better name...
-        self._rules_expand_drvs()
-        self._rules_depluralize_drvs()
+        # Cluster might need to be copied:
+        with DaskContext(self._cluster):
+            self._rules_expand_drvs()
+            self._rules_depluralize_drvs()
 
     def _post_init_populate_rules_with_dimensionless_unit_mappings(self):
         """