@@ -269,11 +269,9 @@ def rec_map_array_container(
269
269
270
270
271
271
def mapped_over_array_containers (
272
- f : Callable [[Any ], Any ],
273
- leaf_class : Optional [type ] = None ) -> Callable [
274
- [ArrayOrContainerT ], ArrayOrContainerT ]:
272
+ f : Callable [[Any ], Any ]) -> Callable [[ArrayOrContainerT ], ArrayOrContainerT ]:
275
273
"""Decorator around :func:`rec_map_array_container`."""
276
- wrapper = partial (rec_map_array_container , f , leaf_class = leaf_class )
274
+ wrapper = partial (rec_map_array_container , f )
277
275
update_wrapper (wrapper , f )
278
276
return wrapper
279
277
@@ -294,13 +292,12 @@ def rec_multimap_array_container(
294
292
295
293
296
294
def multimapped_over_array_containers (
297
- f : Callable [..., Any ],
298
- leaf_class : Optional [type ] = None ) -> Callable [..., Any ]:
295
+ f : Callable [..., Any ]) -> Callable [..., Any ]:
299
296
"""Decorator around :func:`rec_multimap_array_container`."""
300
297
# can't use functools.partial, because its result is insufficiently
301
298
# function-y to be used as a method definition.
302
299
def wrapper (* args : Any ) -> Any :
303
- return rec_multimap_array_container (f , * args , leaf_class = leaf_class )
300
+ return rec_multimap_array_container (f , * args )
304
301
305
302
update_wrapper (wrapper , f )
306
303
return wrapper
0 commit comments