fix docs

Author: crusaderky

docs/api-reference.md

@@ -19,14 +19,3 @@
     sinc
 ```
 
-## Testing utilities
-
-```{eval-rst}
-.. currentmodule:: array_api_extra.testing
-.. autosummary::
-    :nosignatures:
-    :toctree: generated
-
-    lazy_xp_function
-    patch_lazy_xp_functions
-```

docs/index.md

@@ -5,6 +5,7 @@
 :hidden:
 self
 api-reference.md
+testing-utils.md
 contributing.md
 contributors.md
 ```

docs/testing-utils.md

@@ -0,0 +1,14 @@
+# Testing Utilities
+
+These additional functions are meant to be used while unit testing
+Array API compliant packages:
+
+```{eval-rst}
+.. currentmodule:: array_api_extra.testing
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    lazy_xp_function
+    patch_lazy_xp_functions
+```

pixi.lock

@@ -105,6 +105,7 @@ sphinx-autodoc-typehints = "*"
 # Needed to import parsed modules with autodoc
 dask-core = "*"
 pytest = "*"
+typing-extensions = "*"
 
 [tool.pixi.feature.docs.tasks]
 docs = { cmd = "sphinx-build . build/", cwd = "docs" }

pyproject.toml

@@ -111,9 +111,10 @@ def lazy_xp_function(  # type: ignore[no-any-explicit]
       lazy_xp_function(myfunc)
 
       def test_myfunc(xp):
-          a = xp.asarray([1, 2]) # When xp=jax.numpy, this is the same as `b =
-          jax.jit(myfunc)(a)` # When xp=dask.array, crash on compute() or persist() b =
-          myfunc(a)
+          a = xp.asarray([1, 2])
+          # When xp=jax.numpy, this is the same as `b = jax.jit(myfunc)(a)`
+          # When xp=dask.array, crash on compute() or persist() 
+          b = myfunc(a)
 
     Notes
     -----