hide dask-awkward partitionwise import

dask-contrib · Mar 4, 2024 · 941e0b5 · 941e0b5
1 parent eaf1ebe
commit 941e0b5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/dask_histogram/core.py b/src/dask_histogram/core.py
@@ -984,11 +984,12 @@ def _partitioned_histogram_multifill(
     name = f"hist-on-block-{tokenize(data, histref, weights, samples)}"
 
     from dask.base import unpack_collections
-    from dask_awkward.lib.core import partitionwise_layer as dak_pwl
 
     flattened_deps, repacker = unpack_collections(data, weights, samples, histref)
 
     if is_dask_awkward_like(flattened_deps[0]):
+        from dask_awkward.lib.core import partitionwise_layer as dak_pwl
+
         unpacked_multifill = partial(_blocked_multi_dak, repacker)
         graph = dak_pwl(unpacked_multifill, name, *flattened_deps)
     elif is_dataframe_like(flattened_deps[0]):