Skip to content

Commit a6939b1

Browse files
committed
Update
1 parent 15a4e1e commit a6939b1

File tree

2 files changed

+12
-8
lines changed

2 files changed

+12
-8
lines changed

dask_expr/_backends.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,8 @@
44
import pandas as pd
55
from dask.backends import CreationDispatch
66
from dask.dataframe.backends import DataFrameBackendEntrypoint
7-
from dask.dataframe.dispatch import get_parallel_type, to_pandas_dispatch
7+
from dask.dataframe.dispatch import to_pandas_dispatch
88

9-
from dask_expr import FrameBase
109
from dask_expr._dispatch import get_collection_type
1110
from dask_expr._expr import ToBackend
1211

@@ -131,8 +130,3 @@ def get_collection_type_object(_):
131130
@get_collection_type.register_lazy("cudf")
132131
def _register_cudf():
133132
import dask_cudf # noqa: F401
134-
135-
136-
@get_parallel_type.register(FrameBase)
137-
def get_parallel_type_frame(o):
138-
return get_parallel_type(o._meta)

dask_expr/_collection.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,12 @@
3232
is_series_like,
3333
meta_warning,
3434
)
35-
from dask.dataframe.dispatch import is_categorical_dtype, make_meta, meta_nonempty
35+
from dask.dataframe.dispatch import (
36+
get_parallel_type,
37+
is_categorical_dtype,
38+
make_meta,
39+
meta_nonempty,
40+
)
3641
from dask.dataframe.multi import warn_dtype_mismatch
3742
from dask.dataframe.utils import (
3843
AttributeNotImplementedError,
@@ -6517,3 +6522,8 @@ def _compute_partition_stats(
65176522
return (mins, maxes, lens)
65186523
else:
65196524
return (non_empty_mins, non_empty_maxes, lens)
6525+
6526+
6527+
@get_parallel_type.register(FrameBase)
6528+
def get_parallel_type_frame(o):
6529+
return get_parallel_type(o._meta)

0 commit comments

Comments
 (0)