[ci] Fix bug in test_ndarray.py
when using the latest version of JAX
#708
+2
−2
Loading