Uwu

Author: JuliaPoo

Diff for: .github/workflows/array-api-tests-jax.yml

@@ -7,4 +7,4 @@ jobs:
     uses: ./.github/workflows/array-api-tests.yml
     with:
       package-name: jax
-      pytest-extra-args: -k top_k
+      pytest-extra-args: ./array-api-tests/array-api-tests/test_searching_functions.py::test_top_k

Diff for: array_api_compat/jax/__init__.py

@@ -1,5 +1,7 @@
+from jax.numpy import * # quick hack
 from jax import *
 
+
 def top_k(
     x,
     k,