ruff

Author: crusaderky

array_api_compat/common/_helpers.py

@@ -561,7 +561,7 @@ def _jax_namespace(api_version: str | None, use_compat: bool | None) -> Namespac
         # JAX v0.4.32 and newer implements the array API directly in jax.numpy.
         # For older JAX versions, it is available via jax.experimental.array_api.
         # jnp.Array objects gain the __array_namespace__ method.
-        import jax.experimental.array_api
+        import jax.experimental.array_api  # noqa: F401
     # Test api_version
     return jnp.empty(0).__array_namespace__(api_version=api_version)
 

tests/test_array_namespace.py

@@ -54,7 +54,7 @@ def test_jax_backwards_compat():
     jax.experimental.array_api has not been imported yet.
     """
     pytest.importorskip("jax")
-    code = f"""\
+    code = """\
 import sys
 import jax.numpy
 import array_api_compat