Additional validation for the PredictRequest model

predictrequest_final.py

from pydantic import BaseModel, ValidationError, validator

from .ml.model import n_features


class PredictRequest(BaseModel):
    data: List[List[float]]

    @validator("data")
    def check_dimensionality(cls, v):
        for point in v:
            if len(point) != n_features:
                raise ValueError(f"Each data point must contain {n_features} features")

        return v