From 21ba312366cca512844da229cf4fcc1b97eb3b61 Mon Sep 17 00:00:00 2001 From: GALI PREM SAGAR Date: Fri, 18 Nov 2022 16:37:43 -0600 Subject: [PATCH] Fix dask backend dispatch (#12203) This PR fixes a failure being observed in `dask` upstream: https://github.com/dask/dask/issues/9676 Authors: - GALI PREM SAGAR (https://github.com/galipremsagar) Approvers: - Richard (Rick) Zamora (https://github.com/rjzamora) --- python/dask_cudf/dask_cudf/backends.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/python/dask_cudf/dask_cudf/backends.py b/python/dask_cudf/dask_cudf/backends.py index 49b5e725fed..c8e4e015d4a 100644 --- a/python/dask_cudf/dask_cudf/backends.py +++ b/python/dask_cudf/dask_cudf/backends.py @@ -470,7 +470,12 @@ class CudfBackendEntrypoint(DataFrameBackendEntrypoint): @staticmethod def from_dict( - data, npartitions, orient="columns", dtype=None, columns=None + data, + npartitions, + orient="columns", + dtype=None, + columns=None, + constructor=cudf.DataFrame, ): return _default_backend( @@ -480,7 +485,7 @@ def from_dict( orient=orient, dtype=dtype, columns=columns, - constructor=cudf.DataFrame, + constructor=constructor, ) @staticmethod