[ci] Fix bug in test_ndarray.py
when using the latest version of JAX
#1460
Job | Run time |
---|---|
26m 24s | |
28m 38s | |
15m 59s | |
22m 30s | |
10m 15s | |
17m 9s | |
22m 10s | |
9m 14s | |
22m 42s | |
14m 20s | |
9m 11s | |
12m 37s | |
3h 31m 9s |