Skip to content

Commit

Permalink
Avoid dask_cudf.core imports (#5874)
Browse files Browse the repository at this point in the history
Fixes some leftover `dask_cudf.core` imports missed in #5835.

Authors:
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Richard (Rick) Zamora (https://github.com/rjzamora)
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: #5874
  • Loading branch information
bdice authored Apr 29, 2024
1 parent 6d3bb0d commit de181d8
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
6 changes: 3 additions & 3 deletions python/cuml/internals/array.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,8 @@
CudfDataFrame = gpu_only_import_from("cudf", "DataFrame")
CudfIndex = gpu_only_import_from("cudf", "Index")
CudfSeries = gpu_only_import_from("cudf", "Series")
DaskCudfDataFrame = gpu_only_import_from("dask_cudf.core", "DataFrame")
DaskCudfSeries = gpu_only_import_from("dask_cudf.core", "Series")
DaskCudfDataFrame = gpu_only_import_from("dask_cudf", "DataFrame")
DaskCudfSeries = gpu_only_import_from("dask_cudf", "Series")
DaskDataFrame = gpu_only_import_from("dask.dataframe", "DataFrame")
DaskSeries = gpu_only_import_from("dask.dataframe", "Series")
DeviceBuffer = gpu_only_import_from("rmm", "DeviceBuffer")
Expand Down Expand Up @@ -793,7 +793,7 @@ def deserialize(cls, header: dict, frames: list):

if header["desc"]["shape"] != ary._array_interface["shape"]:
raise ValueError(
f"Received a `Buffer` with the wrong size."
"Received a `Buffer` with the wrong size."
f" Expected {header['desc']['shape']}, "
f"but got {ary._array_interface['shape']}"
)
Expand Down
6 changes: 3 additions & 3 deletions python/cuml/internals/input_utils.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2019-2023, NVIDIA CORPORATION.
# Copyright (c) 2019-2024, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -46,8 +46,8 @@
cp_ndarray = gpu_only_import_from("cupy", "ndarray")
CudfSeries = gpu_only_import_from("cudf", "Series")
CudfDataFrame = gpu_only_import_from("cudf", "DataFrame")
DaskCudfSeries = gpu_only_import_from("dask_cudf.core", "Series")
DaskCudfDataFrame = gpu_only_import_from("dask_cudf.core", "DataFrame")
DaskCudfSeries = gpu_only_import_from("dask_cudf", "Series")
DaskCudfDataFrame = gpu_only_import_from("dask_cudf", "DataFrame")
np_ndarray = cpu_only_import_from("numpy", "ndarray")
numba_devicearray = gpu_only_import_from("numba.cuda", "devicearray")
try:
Expand Down

0 comments on commit de181d8

Please sign in to comment.