lint

Author: crusaderky

docs/api-reference.md

@@ -18,4 +18,3 @@
     setdiff1d
     sinc
 ```
-

docs/testing-utils.md

@@ -1,7 +1,7 @@
 # Testing Utilities
 
-These additional functions are meant to be used while unit testing
-Array API compliant packages:
+These additional functions are meant to be used while unit testing Array API
+compliant packages:
 
 ```{eval-rst}
 .. currentmodule:: array_api_extra.testing
@@ -11,4 +11,4 @@ Array API compliant packages:
 
     lazy_xp_function
     patch_lazy_xp_functions
-```
+```

src/array_api_extra/testing.py

@@ -87,8 +87,8 @@ def lazy_xp_function(  # type: ignore[no-any-explicit]
         graph.
     jax_jit : bool, optional
         Set to True to replace `func` with ``jax.jit(func)`` after calling the
-        :func:`patch_lazy_xp_functions` test helper with ``xp=jax.numpy``. Set to False if
-        `func` is only compatible with eager (non-jitted) JAX. Default: True.
+        :func:`patch_lazy_xp_functions` test helper with ``xp=jax.numpy``. Set to False
+        if `func` is only compatible with eager (non-jitted) JAX. Default: True.
     static_argnums : int | Sequence[int], optional
         Passed to jax.jit. Positional arguments to treat as static (compile-time
         constant). Default: infer from `static_argnames` using
@@ -113,7 +113,7 @@ def lazy_xp_function(  # type: ignore[no-any-explicit]
       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() 
+          # When xp=dask.array, crash on compute() or persist()
           b = myfunc(a)
 
     Notes
@@ -150,8 +150,8 @@ def patch_lazy_xp_functions(
     :func:`lazy_xp_function` in the globals of the module that defines the current test
     and wrap them with :func:`jax.jit`. Unwrap them at the end of the test.
 
-    If ``xp==dask.array``, wrap the functions with a decorator that disables ``compute()``
-    and ``persist()``.
+    If ``xp==dask.array``, wrap the functions with a decorator that disables
+    ``compute()`` and ``persist()``.
 
     This function should be typically called by your library's `xp` fixture that runs
     tests on multiple backends::