File tree Expand file tree Collapse file tree 1 file changed +3
-8
lines changed Expand file tree Collapse file tree 1 file changed +3
-8
lines changed Original file line number Diff line number Diff line change @@ -295,16 +295,11 @@ def _apply_lazy_wrapper( # type: ignore[no-any-explicit] # numpydoc ignore=PR0
295
295
"""
296
296
Helper of `apply_lazy`.
297
297
298
- Given a function that accepts one or more numpy arrays as positional arguments and
299
- returns a single numpy array or a sequence of numpy arrays, return a function that
300
- accepts the same number of Array API arrays and always returns a tuple of Array API
301
- array.
298
+ Given a function that accepts one or more arrays as positional arguments and returns
299
+ a single array-like or a sequence of array-likes, return a function that accepts the
300
+ same number of Array API arrays and always returns a tuple of Array API array.
302
301
303
302
Any keyword arguments are passed through verbatim to the wrapped function.
304
-
305
- Raise if np.asarray raises on any input. This typically happens if the input is lazy
306
- and has a guard against being implicitly turned into a NumPy array (e.g.
307
- densification for sparse arrays, device->host transfer for cupy and torch arrays).
308
303
"""
309
304
310
305
# On Dask, @wraps causes the graph key to contain the wrapped function's name
You can’t perform that action at this time.
0 commit comments