Pydantic numpy ndarray type

# Define custom type validatable by Pydantic.
class TypedArray(numpy.ndarray):
    @classmethod
    def __get_validators__(cls):
        yield cls.validate_type

    @classmethod
    def validate_type(cls, val):
        return numpy.array(val, dtype=cls.inner_type)

class ArrayMeta(type):
    def __getitem__(self, t):
        return type('Array', (TypedArray,), {'inner_type': t})

class Array(numpy.ndarray, metaclass=ArrayMeta):
    pass

class Model(BaseModel):
    values: Array[float]

m = Model(values=[1, 2, 3])
print(m)

Source: github.com

Similar codes

21 votes280 views5 answers
Tkinter window
Python
0 vote311 views1 answer
Python convert png to jpg
Python
5 votes332 views3 answers
Write to file python 3
Python
0 vote314 views1 answer
Read file from s3 python
Python
0 vote318 views1 answer
Deques in python3
Python
14 votes334 views6 answers
Join tables pandas
Python
1 vote411 views1 answer
Django queryset group by count
Python Django
0 vote309 views1 answer
Pandas filter on range of values
Python
-1 vote312 views1 answer
Write Python Program to find the square root of an input number. (create a user-define function of square root).
Python
0 vote311 views1 answer
Pandas dataframe any along row
Python