Skip to content

Commit 69e0e35

Browse files
committed
Fix CI
1 parent 97d9063 commit 69e0e35

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

pandas/core/apply.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -209,9 +209,12 @@ def apply(
209209
"""
210210
Apply `func` along the given axis using Numba.
211211
"""
212+
engine_kwargs: dict[str, bool] | None = (
213+
decorator if isinstance(decorator, dict) else None
214+
)
212215

213216
looper_args, looper_kwargs = prepare_function_arguments(
214-
func, # type: ignore[arg-type]
217+
func,
215218
args,
216219
kwargs,
217220
num_required_args=1,
@@ -221,8 +224,8 @@ def apply(
221224
# [..., Any] | str] | dict[Hashable,Callable[..., Any] | str |
222225
# list[Callable[..., Any] | str]]"; expected "Hashable"
223226
nb_looper = generate_apply_looper(
224-
func, # type: ignore[arg-type]
225-
**get_jit_arguments(decorator),
227+
func,
228+
**get_jit_arguments(engine_kwargs),
226229
)
227230
result = nb_looper(data, axis, *looper_args)
228231
# If we made the result 2-D, squeeze it back to 1-D

0 commit comments

Comments
 (0)