Add jaxtyping as an optional dependency, make sure typ is jaxtyping and not numpy

Author: pablovela5620

pyproject.toml

@@ -38,6 +38,7 @@ numpy = [
   { version = ">1.22.0,<2.0.0", markers = "python_version ~= '3.11' and (extra == 'numpy' or extra == 'all')", optional = true },
   { version = ">1.22.0,<2.0.0", markers = "python_version ~= '3.12' and (extra == 'numpy' or extra == 'all')", optional = true },
 ]
+jaxtyping = { version = "*", markers = "extra == 'jaxtyping' or extra == 'all'", optional = true }
 orjson = { version = "*", markers = "extra == 'orjson' or extra == 'all'", optional = true }
 plum-dispatch = ">=2,<2.3"
 beartype = ">=0.18.4"

serde/numpy.py

@@ -78,7 +78,7 @@ def is_numpy_jaxtyping(typ) -> bool:
             origin = get_origin(typ)
             if origin is not None:
                 typ = origin
-            return issubclass(typ, np.ndarray)
+            return typ is not np.ndarray and issubclass(typ, np.ndarray)
         except TypeError:
             return False