WIP apply_numpy_func

Author: crusaderky

.pre-commit-config.yaml

@@ -59,13 +59,13 @@ repos:
     hooks:
       - id: shellcheck
 
-  - repo: local
-    hooks:
-      - id: disallow-caps
-        name: Disallow improper capitalization
-        language: pygrep
-        entry: PyBind|Numpy|Cmake|CCache|Github|PyTest
-        exclude: .pre-commit-config.yaml
+  # - repo: local
+  #   hooks:
+  #     - id: disallow-caps
+  #       name: Disallow improper capitalization
+  #       language: pygrep
+  #       entry: PyBind|Numpy|Cmake|CCache|Github|PyTest
+  #       exclude: .pre-commit-config.yaml
 
   - repo: https://github.com/abravalheri/validate-pyproject
     rev: "v0.23"

docs/api-reference.md

@@ -6,6 +6,7 @@
     :nosignatures:
     :toctree: generated
 
+    apply_numpy_func
     at
     atleast_nd
     cov

docs/conf.py

@@ -54,6 +54,7 @@
 intersphinx_mapping = {
     "python": ("https://docs.python.org/3", None),
     "jax": ("https://jax.readthedocs.io/en/latest", None),
+    "dask": ("https://docs.dask.org/en/stable", None),
 }
 
 nitpick_ignore = [

src/array_api_extra/__init__.py

@@ -1,5 +1,6 @@
 """Extra array functions built on top of the array API standard."""
 
+from ._apply import apply_numpy_func
 from ._funcs import (
     at,
     atleast_nd,
@@ -17,6 +18,7 @@
 # pylint: disable=duplicate-code
 __all__ = [
     "__version__",
+    "apply_numpy_func",
     "at",
     "atleast_nd",
     "cov",