p2p-ld/numpydantic
1
0
Fork 0
mirror of https://github.com/p2p-ld/numpydantic.git synced 2024-11-14 02:34:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

python 3.9 compat

Browse source
This commit is contained in:
sneakers-the-rat 2024-09-23 23:30:40 -07:00
parent 85cef50603
commit e63d9268b1
Signed by untrusted user who does not match committer: jonny
GPG key ID: 6DCB96EF1E4D232D
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/numpydantic/validation/dtype.py
View file

@ -3,14 +3,18 @@ Helper functions for validation of dtype.
For literal dtypes intended for use by end-users, see :mod:`numpydantic.dtype`
"""
from types import UnionType
import sys
from typing import Any, Union, get_args, get_origin
import numpy as np
from numpydantic.types import DtypeType
if sys.version_info >= (3, 10):
from types import UnionType
else:
UnionType = None
def validate_dtype(dtype: Any, target: DtypeType) -> bool:
"""
@ -52,4 +56,7 @@ def is_union(dtype: DtypeType) -> bool:
"""
Check if a dtype is a union
"""
return get_origin(dtype) in (Union, UnionType)
if UnionType is None:
return get_origin(dtype) is Union
else:
return get_origin(dtype) in (Union, UnionType)