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(
209
209
"""
210
210
Apply `func` along the given axis using Numba.
211
211
"""
212
+ engine_kwargs : dict [str , bool ] | None = (
213
+ decorator if isinstance (decorator , dict ) else None
214
+ )
212
215
213
216
looper_args , looper_kwargs = prepare_function_arguments (
214
- func , # type: ignore[arg-type]
217
+ func ,
215
218
args ,
216
219
kwargs ,
217
220
num_required_args = 1 ,
@@ -221,8 +224,8 @@ def apply(
221
224
# [..., Any] | str] | dict[Hashable,Callable[..., Any] | str |
222
225
# list[Callable[..., Any] | str]]"; expected "Hashable"
223
226
nb_looper = generate_apply_looper (
224
- func , # type: ignore[arg-type]
225
- ** get_jit_arguments (decorator ),
227
+ func ,
228
+ ** get_jit_arguments (engine_kwargs ),
226
229
)
227
230
result = nb_looper (data , axis , * looper_args )
228
231
# 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