from pydantic import BaseModel, Field
from typing import Optional


class ProjectTrainOut(BaseModel):
    """项目训练版本信息表"""
    id: Optional[int] = Field(None, description="训练id")
    train_version: Optional[str] = Field(None, description="训练版本号")
    best_pt: Optional[str] = Field(None, description="最好")
    last_pt: Optional[str] = Field(None, description="最后")

    class Config:
        orm_mode = True