Skip to content

Commit

Permalink
Merge branch 'maybe_fix_tree' of https://github.com/martindurant/dask…
Browse files Browse the repository at this point in the history
…-histogram into maybe_fix_tree
  • Loading branch information
martindurant committed Jan 27, 2025
2 parents 298994e + 19e0cd4 commit f4b5d93
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions src/dask_histogram/layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,17 @@
from dask.layers import DataFrameTreeReduction
except ImportError:
try:
from dask_awkward.layers import AwkwardTreeReductionLayer as DataFrameTreeReduction
from dask_awkward.layers import (
AwkwardTreeReductionLayer as DataFrameTreeReduction,
)
except ImportError:
DataFrameTreeReduction = None

if DataFrameTreeReduction is None:
raise ImportError("DataFrameReduction is unimportable - either downgrade dask to <2025"
"or install dask-awkward >=2025.")
raise ImportError(
"DataFrameReduction is unimportable - either downgrade dask to <2025"
"or install dask-awkward >=2025."
)


class MockableDataFrameTreeReduction(DataFrameTreeReduction):
Expand Down

0 comments on commit f4b5d93

Please sign in to comment.