File size: 532 Bytes
ab8b5a4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
"""
Schema for the inference endpoint
"""
from pydantic import BaseModel, validator
from enum import Enum
class Sex(str, Enum):
male = "m"
female = "f"
class Patient(BaseModel):
age: int
sex: Sex
@validator('age')
def patient_age_must_be_legit(cls, v):
if v < 18:
raise ValueError('age must be at least 18')
if v > 100:
raise ValueError('age must be at most 100')
return v
class InferenceRequest(BaseModel):
model_name: str
patient: Patient
|