From 7fee64ada60726c1193aa247777a5d952bebf844 Mon Sep 17 00:00:00 2001 From: sneakers-the-rat Date: Fri, 17 May 2024 16:27:05 -0700 Subject: [PATCH] don't try and access __name__ on typing classes --- src/numpydantic/ndarray.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/numpydantic/ndarray.py b/src/numpydantic/ndarray.py index 3f205b1..bcb35bd 100644 --- a/src/numpydantic/ndarray.py +++ b/src/numpydantic/ndarray.py @@ -140,7 +140,10 @@ class NDArray(NPTypingType, metaclass=NDArrayMeta): json_schema = handler.resolve_ref_schema(json_schema) dtype = cls.__args__[1] - if not isinstance(dtype, tuple) and dtype.__module__ != "builtins": + if not isinstance(dtype, tuple) and dtype.__module__ not in ( + "builtins", + "typing", + ): json_schema["dtype"] = ".".join([dtype.__module__, dtype.__name__]) return json_schema