Merge pull request #506 from itamarst/505-incompatibility-with-dask-2024112

Fix incompatibility with newer Dask

Author: itamarst

docs/source/news.rst

@@ -8,6 +8,10 @@ Enhancements:
 
 * Added official support for Python 3.13, though 1.15.0 works fine too.
 
+Changes:
+
+* Tests pass with latest Dask.
+
 1.15.0
 ^^^^^^
 

eliot/dask.py

@@ -6,6 +6,7 @@
 
 try:
     from dask.distributed import Future
+    from dask.highlevelgraph import HighLevelGraph
 except:
 
     class Future(object):
@@ -108,6 +109,9 @@ def _add_logging(dsk, ignore=None):
 
     @return: New Dask graph.
     """
+    if isinstance(dsk, HighLevelGraph):
+        dsk = dsk.to_dict()
+
     ctx = current_action()
     result = {}