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 vote234 views1 answer
Mad libs game prompt python
Python
7 votes288 views2 answers
Cv2 load image
Python
7 votes199 views2 answers
Pandas read_csv ignore unnamed columns
Python
0 vote173 views1 answer
Nested For loop in Python
Python
8 votes199 views3 answers
Convert ('a', 'b'): 1 dictionary to a dataframe
Python
0 vote197 views1 answer
Split the string using the default arguments
Python
32 votes258 views6 answers
Merge pandas
Python
0 vote202 views1 answer
Pandas increment value on condition
Python
11 votes236 views5 answers
Change default python version mac
Python
78 votes185 views6 answers
Python check version
Python