simple enough fix - handle typing.UnionType in additional dtype m…

…etadata in json schema
p2p-ld · Dec 14, 2024 · 47d8039 · 47d8039
1 parent a02d9b2
commit 47d8039
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/numpydantic/ndarray.py b/src/numpydantic/ndarray.py
@@ -204,9 +204,15 @@ def __get_pydantic_json_schema__(
         json_schema = handler(schema["metadata"])
         json_schema = handler.resolve_ref_schema(json_schema)
 
-        if not isinstance(dtype, tuple) and dtype.__module__ not in (
-            "builtins",
-            "typing",
+        if (
+            not isinstance(dtype, tuple)
+            and dtype.__module__
+            not in (
+                "builtins",
+                "typing",
+                "types",
+            )
+            and hasattr(dtype, "__name__")
         ):
             json_schema["dtype"] = ".".join([dtype.__module__, dtype.__name__])