File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -189,7 +189,7 @@ def map(
189
189
func ,
190
190
args : tuple ,
191
191
kwargs : dict ,
192
- engine_kwargs : dict | None ,
192
+ decorator : Callable | None ,
193
193
skip_na : bool ,
194
194
):
195
195
"""
@@ -203,7 +203,7 @@ def apply(
203
203
func ,
204
204
args : tuple ,
205
205
kwargs : dict ,
206
- engine_kwargs : dict | None ,
206
+ decorator : Callable ,
207
207
axis : int | str ,
208
208
):
209
209
"""
@@ -222,7 +222,7 @@ def apply(
222
222
# list[Callable[..., Any] | str]]"; expected "Hashable"
223
223
nb_looper = generate_apply_looper (
224
224
func , # type: ignore[arg-type]
225
- ** get_jit_arguments (engine_kwargs ),
225
+ ** get_jit_arguments (decorator ),
226
226
)
227
227
result = nb_looper (data , axis , * looper_args )
228
228
# 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