• "numpy.core.multiarray._reconstruct",
• "jax._src.array._reconstruct_array",
• "numpy.dtype",
• "numpy.ndarray"

How to fix it?