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

97 votes223 views6 answers
Download python
Python
5 votes299 views2 answers
How to set icon in tkinter
Python
1 vote216 views1 answer
Python how to set multiple conditional for single var
Python
0 vote270 views1 answer
Pandas resamples stratified by columns values
Python
0 vote231 views1 answer
Convert json file to dict - if comming as list
Python
0 vote264 views2 answers
Merge two sorted lists into one sorted list
Python
1 vote308 views1 answer
Insert column at specific position in pandas dataframe
Python
2 votes249 views1 answer
Readline in a file
Python
0 vote237 views1 answer
Python test if string begins with python
Python
6 votes237 views1 answer
Axis number size matplotlib
Python