return only array shapes and types in Error?

[bkmi]

First of all, awesome package.

Second, is there a flag so that the errors returned only return the shapes and dtypes of every array? I am typically typechecking huge arrays and I get an extremely long error message back printing the entire array. This seems unnecessary to me. How can I change it?

I'm using beartype as a backend.

[patrick-kidger]

Aha! This is timely.

Can you try installing jaxtyping from HEAD? I've just added this, would be awesome to have someone give it a try in advance of the next release :)

[bkmi]

I can give it a try after the icml deadline ;)