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
295295 """
296296 Helper of `apply_lazy`.
297297
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.
302301
303302 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).
308303 """
309304
310305 # 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