Skip to content

Commit 176753b

Browse files
committed
Updated executor to accept decorator
1 parent bc2939b commit 176753b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas/core/_numba/executor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,14 @@
1818

1919

2020
@functools.cache
21-
def generate_apply_looper(func, nopython=True, nogil=True, parallel=False):
21+
def generate_apply_looper(func, decorator: Callable):
2222
if TYPE_CHECKING:
2323
import numba
2424
else:
2525
numba = import_optional_dependency("numba")
2626
nb_compat_func = jit_user_function(func)
2727

28-
@numba.jit(nopython=nopython, nogil=nogil, parallel=parallel)
28+
@decorator # type: ignore
2929
def nb_looper(values, axis, *args):
3030
# Operate on the first row/col in order to get
3131
# the output shape

0 commit comments

Comments
 (0)