Merge pull request #152 from dask-contrib/blank

[fix] import dak if it exists
dask-contrib · Sep 19, 2024 · 8923899 · 8923899
2 parents fb37983 + 6f1b277
commit 8923899
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/afile b/afile
diff --git a/src/dask_histogram/core.py b/src/dask_histogram/core.py
@@ -524,10 +524,12 @@ def _get_optimization_function():
     # running this optimization even in cases where it's unncessary
     # because if no AwkwardInputLayers from dask-awkward are
     # detected then the original graph is returned unchanged.
-    if dask.config.get("awkward", default=False):
+    try:
         from dask_awkward.lib.optimize import all_optimizations
 
         return all_optimizations
+    except (ImportError, ModuleNotFoundError):
+        pass
     return optimize