Skip to content

Commit 92aee7b

Browse files
pfackeldeyalexander-held
authored andcommitted
compat with older coffea versions
1 parent bef6bae commit 92aee7b

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

util/_dask.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
from coffea.nanoevents import NanoEventsFactory
4040
from coffea.processor import Accumulatable, accumulate
4141
from coffea.processor.executor import WorkItem
42-
from coffea.util import coffea_console, rich_bar
42+
from coffea.util import rich_bar
4343
from dask.distributed import Client
4444
from dask.tokenize import tokenize
4545
from rich.console import Group
@@ -48,6 +48,14 @@
4848

4949
from util._futures import DynamicAsCompleted, FutureLike
5050

51+
try:
52+
from coffea.util import coffea_console
53+
except ImportError:
54+
from rich.console import Console
55+
56+
coffea_console = Console()
57+
58+
5159
_processing_sentinel = object()
5260
_final_merge_sentinel = object()
5361

0 commit comments

Comments
 (0)