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

4 votes191 views2 answers
Python convert nan to empty string
Python
0 vote197 views1 answer
How to serach for multiple attributes in xpath selenium python
Python
0 vote191 views1 answer
Python type hints list of class
Python
0 vote190 views2 answers
Keras name layers
Python
36 votes190 views4 answers
Python print string in red color
Python
20 votes273 views3 answers
Download image python
Python
1 vote190 views1 answer
Object.image.url email template django
Python
1 vote229 views2 answers
Get the torch version
Python
114 votes180 views6 answers
Amc price
Python
6 votes194 views1 answer
Copy file in python3
Python