Closed pablovela5620 closed 1 month ago
Hi @pablovela5620! jaxtyping is not supported by pyserde. Sorry I am not a user of jaxtyping/numpy, I can not help. If you're interested in contributing, I am happy to assist you though.
That makes sense @yukinarit ! Id be happy to help, I've already tried looking through the code some. If you could point me in the right direction I can try to add support. Since the underlying type is np.ndarray I'm guessing it shouldn't be that difficult
Sure @pablovela5620
As you may already know, serialize/deserialize code for numpy can be found in numpy.py
You may need to modify se.py and de.py. Those two source files are about codegen.
If you have any questions, feel free to ask anything. @kmsquire is the author of the numpy support. Would be nice if he could also help us.
I'm wondering if pyserde support jaxtyping and this is a bug, or if it is just unsupported. The following works fine
where as if I use a jaxtyping type (for dtype and shape checking with beartype) I get an error