chore(api): add mypy coverage to apis

Signed-off-by: Mustafa Elbehery <melbeher@redhat.com>
This commit is contained in:
Mustafa Elbehery 2025-07-08 15:53:35 +02:00
parent 7f3661e7d8
commit 21afea8412
4 changed files with 101 additions and 5 deletions

View file

@ -19,8 +19,10 @@ class PostTrainingMetric(BaseModel):
perplexity: float
@json_schema_type(schema={"description": "Checkpoint created during training runs"})
@json_schema_type
class Checkpoint(BaseModel):
"""Checkpoint created during training runs"""
identifier: str
created_at: datetime
epoch: int