Dask moved public objects in 2024.9.0 (#787)

bluesky · Sep 19, 2024 · dd8d8f1 · dd8d8f1
1 parent 337ebab
commit dd8d8f1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,10 @@ Write the date in place of the "Unreleased" in the case a new version is release
 
 - Added support for explicit units in numpy datetime64 dtypes.
 
+### Fixed
+
+- Adapt to change in dask public API in dask 2024.9.0.
+
 ## v0.1.0b8 (2024-09-06)
 
 ### Fixed

diff --git a/tiled/server/etag.py b/tiled/server/etag.py
@@ -1,5 +1,12 @@
-from dask.base import normalize_object, normalize_token
-from dask.base import tokenize as dask_tokenize
+import importlib.util
+
+if importlib.util.find_spec("dask.tokenize"):
+    # moved in dask version 2024.9.0
+    from dask.tokenize import normalize_object, normalize_token
+    from dask.tokenize import tokenize as dask_tokenize
+else:
+    from dask.base import normalize_object, normalize_token
+    from dask.base import tokenize as dask_tokenize
 
 
 def tokenize(obj):