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

0 vote175 views1 answer
How to set up pygame
Python
2 votes182 views1 answer
Save file python tkinter
Python
0 vote184 views1 answer
Count number of subdirectories
Python
42 votes176 views6 answers
Python regex
Python
3 votes178 views4 answers
Csv to python dictionary
Python
2 votes161 views1 answer
Pandas series select first value
Python
0 vote181 views1 answer
Port python script to jupyter notebook
Python
0 vote159 views1 answer
Python tf.maximum
Python
0 vote216 views1 answer
Check even or odd in single line
Python
15 votes195 views3 answers
NAN values count python
Python