Skip to content

Commit

Permalink
[REVIEW] Add make_meta_dispatch handling (#637)
Browse files Browse the repository at this point in the history
* add make_meta handling

* black
  • Loading branch information
galipremsagar authored Jun 3, 2021
1 parent 236d5aa commit 1432d1f
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion dask_cuda/proxy_object.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,11 @@
except ImportError:
from dask.dataframe.methods import concat_pandas

try:
from dask.dataframe.dispatch import make_meta_dispatch as make_meta_dispatch
except ImportError:
from dask.dataframe.utils import make_meta as make_meta_dispatch

from .get_device_memory_objects import get_device_memory_objects
from .is_device_object import is_device_object

Expand Down Expand Up @@ -732,7 +737,7 @@ def wrapper(*args, **kwargs):
# Register dispatch of ProxyObject on all known dispatch objects
for dispatch in (
dask.dataframe.core.hash_object_dispatch,
dask.dataframe.utils.make_meta,
make_meta_dispatch,
dask.dataframe.utils.make_scalar,
dask.dataframe.core.group_split_dispatch,
dask.array.core.tensordot_lookup,
Expand Down

0 comments on commit 1432d1f

Please sign in to comment.