diff --git a/src/array_api_extra/_apply.py b/src/array_api_extra/_apply.py index 6d792d5..ce03346 100644 --- a/src/array_api_extra/_apply.py +++ b/src/array_api_extra/_apply.py @@ -39,8 +39,7 @@ def apply_numpy_func( # type: ignore[no-any-explicit] **kwargs: Any, ) -> tuple[Array, ...]: """ - Apply a function that operates on NumPY arrays to any Array API compliant arrays, - as long as you can apply ``np.asarray`` to them. + Apply a function that operates on NumPY arrays to Array API compliant arrays. Parameters ---------- @@ -52,7 +51,9 @@ def apply_numpy_func( # type: ignore[no-any-explicit] It must be a pure function, i.e. without side effects such as disk output, as depending on the backend it may be executed more than once. *args : Array - One or more Array API compliant arrays. + One or more Array API compliant arrays. You need to be able to apply + ``np.asarray()`` to them to convert them to numpy; read notes below about + specific backends. shapes : Sequence[tuple[int, ...]], optional Sequence of output shapes, one for each output of `func`. If `func` returns a single (non-sequence) output, this must be a sequence