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

5 votes245 views3 answers
Print complete dataframe pandas
Python
3 votes285 views3 answers
Import flask
Python
0 vote266 views1 answer
Convert to pdf fresh little library that outputs our notebook in a nice LaTex format without installing/doing anything else.
Python
0 vote274 views1 answer
Pillow python update image
Python
2 votes249 views2 answers
Import seaborn
Python
8 votes262 views3 answers
Check pandas version
Python
0 vote258 views2 answers
Split column in exact spot python
Python
0 vote283 views1 answer
Request.args.get check if defined
Python
44 votes249 views3 answers
How can I sort a dictionary in python according to its values?
Python
0 vote243 views1 answer
Python numpy array check if all nans
Python