File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -1023,9 +1023,12 @@ def log(self, base: float) -> Self:
10231023 dtype_backend = get_dtype_backend (native .dtype , implementation = implementation )
10241024
10251025 if implementation .is_cudf ():
1026- import cupy as cp
1026+ import cupy as cp # ignore-banned-import # cuDF dependency.
10271027
1028- return self ._with_native (cp .log (native ) / cp .log (base ))
1028+ native = self .native
1029+ log_arr = cp .log (native ) / cp .log (base )
1030+ result_native = type (native )(log_arr , index = native .index , name = native .name )
1031+ return self ._with_native (result_native )
10291032
10301033 if dtype_backend == "pyarrow" :
10311034 import pyarrow .compute as pc
Original file line number Diff line number Diff line change 55
66BANNED_IMPORTS = {
77 "cudf" ,
8+ "cupy" ,
89 "dask" ,
910 "dask.dataframe" ,
1011 "dask_expr" ,
You can’t perform that action at this time.
0 commit comments