File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments