docs updates

Author: lucascolley

docs/conf.py

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

pyproject.toml

@@ -309,6 +309,7 @@ checks = [
     "ES01",
 ]
 exclude = [ # don't report on objects that match any of these regex
+    '.*test_at.*',
     '.*test_funcs.*',
     '.*test_utils.*',
     '.*test_version.*',

src/array_api_extra/_funcs.py

@@ -566,7 +566,7 @@ class at:  # pylint: disable=invalid-name  # numpydoc ignore=PR02
     x : array
         Input array.
     idx : index, optional
-        Only `Array API compliant indices
+        Only `array API standard compliant indices
         <https://data-apis.org/array-api/latest/API_specification/indexing.html>`_
         are supported.
 
@@ -623,8 +623,7 @@ class at:  # pylint: disable=invalid-name  # numpydoc ignore=PR02
 
     See Also
     --------
-    jax.numpy.ndarray.at : `docs
-      <https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ndarray.at.html>`_.
+    jax.numpy.ndarray.at : Equivalent array method in JAX.
 
     Notes
     -----

src/array_api_extra/_lib/_compat.pyi

@@ -4,6 +4,8 @@ from types import ModuleType
 
 from ._typing import Array, Device
 
+# pylint: disable=missing-class-docstring,unused-argument
+
 class ArrayModule(ModuleType):
     def device(self, x: Array, /) -> Device: ...