fix ci?

Author: lithomas1

.github/workflows/tests.yml

@@ -15,7 +15,7 @@ jobs:
       - name: Install Dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest numpy torch dask[array] jax[cpu]>=0.4.24
+          python -m pip install pytest numpy torch dask[array] jax[cpu]
 
       - name: Run Tests
         run: |

tests/_helpers.py

@@ -8,7 +8,7 @@
 def import_(library, wrapper=False):
     if library == 'cupy':
         return pytest.importorskip(library)
-    if 'jax' in library and sys.version_info <= (3, 8):
+    if 'jax' in library and sys.version_info < (3, 9):
         pytest.skip('JAX array API support does not support Python 3.8')
 
     if wrapper: