File tree Expand file tree Collapse file tree 2 files changed +12
-8
lines changed
Expand file tree Collapse file tree 2 files changed +12
-8
lines changed Original file line number Diff line number Diff line change 44import pandas as pd
55from dask .backends import CreationDispatch
66from 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
109from dask_expr ._dispatch import get_collection_type
1110from dask_expr ._expr import ToBackend
1211
@@ -131,8 +130,3 @@ def get_collection_type_object(_):
131130@get_collection_type .register_lazy ("cudf" )
132131def _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 )
Original file line number Diff line number Diff line change 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+ )
3641from dask .dataframe .multi import warn_dtype_mismatch
3742from 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 )
You can’t perform that action at this time.
0 commit comments