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

2 votes141 views3 answers
How to copy items from 1 list to another in python
Python
0 vote134 views1 answer
Production mode flask
Python
0 vote131 views1 answer
Python lvl up
Python
0 vote138 views1 answer
How to group multiple test in pytest
Python
0 vote127 views1 answer
Shared SHMEM python
Python
0 vote162 views1 answer
Vvm 2020 exam date
Python
9 votes139 views1 answer
Find nan value in dataframe python
Python
26 votes152 views6 answers
Convert list of strings to ints python
Python
18 votes134 views3 answers
Change specific column name pandas
Python
3 votes136 views2 answers
Python subprocess exception handling
Python