Fix a _check_arraylike error.
Fix a _check_arraylike error.

Before, if `repeats` was the the non-array arg the error would display
the wrong arg position (position 0 -> position 1)

google/jax,google/jax,google/jax,google/jax