diff --git a/src/pytypes/onefuzztypes/models.py b/src/pytypes/onefuzztypes/models.py index 6c82bf714..39399434d 100644 --- a/src/pytypes/onefuzztypes/models.py +++ b/src/pytypes/onefuzztypes/models.py @@ -562,6 +562,12 @@ class Scaleset(BaseModel): client_object_id: Optional[UUID] tags: Dict[str, str] = Field(default_factory=lambda: {}) + @validator("size", allow_reuse=True) + def check_size(cls, value: int) -> int: + if value < 0: + raise ValueError("Invalid scaleset size") + return value + class NotificationConfig(BaseModel): config: NotificationTemplate