wip: more log messages in context manager

esm-tools · Jan 9, 2025 · 097c250 · 097c250
1 parent 4d40d82
commit 097c250
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/pymorize/cluster.py b/src/pymorize/cluster.py
@@ -57,10 +57,12 @@ class DaskContext:
     @classmethod
     @contextmanager
     def set_cluster(cls, cluster):
+        logger.info(f"Setting Dask cluster {cluster=} in context!")
         cls._current_cluster = cluster
         try:
             yield
         finally:
+            logger.info("Removing Dask cluster from context!")
             cls._current_cluster = None
 
     @classmethod