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 vote264 views1 answer
Add to middle of string python
Python
0 vote273 views1 answer
Python how to turn a word into a list
Python
0 vote253 views1 answer
Python: if null give a value if not null concatenate
Python
2 votes245 views1 answer
Python only numbers in string
Python
0 vote241 views1 answer
Python how to dump exception stak
Python
5 votes252 views2 answers
F string decimal places
Python
0 vote248 views1 answer
Pip ne marche pas
Python
0 vote259 views1 answer
Python apply function to list
Python
0 vote251 views1 answer
Convert integer unix to timestamp python
Python
2 votes261 views1 answer
Torch device
Python