[ci] Fix bug in `test_ndarray.py` when using the latest version of JAX#708

Merged

chaoming0625 merged 2 commits intomasterfrom fix-ci

Dec 16, 2024

+2-2

Commits on Dec 15, 2024

Update test_ndarray.py
Routhleck
committed

Commits on Dec 16, 2024

Update ndarray.py
Routhleck
committed