完成项目训练模块的接口测试

apps/business/train/schemas/project_train.py

@@ -6,7 +6,8 @@
 # @IDE            : PyCharm
 # @desc           : pydantic 模型，用于数据库序列化操作
 
-from datetime import datetime
+from core.data_types import DatetimeStr
+
 from pydantic import BaseModel, Field, ConfigDict
 from typing import Optional
 
@@ -18,6 +19,7 @@ from typing import Optional
 class ProjectTrainIn(BaseModel):
     project_id: Optional[int] = Field(..., description="项目id")
     weights_id: Optional[str] = Field(None, description="权重文件")
+    weights_name: Optional[str] = Field(None, description="权重文件名称")
     epochs: Optional[int] = Field(50, description="训练轮数")
     patience: Optional[int] = Field(20, description="早停的耐心值")
 
@@ -28,7 +30,7 @@ class ProjectTrainOut(BaseModel):
     weights_name: Optional[str] = Field(None, description="权重名称")
     epochs: Optional[int] = Field(None, description="训练轮数")
     patience: Optional[int] = Field(None, description="早停的耐心值")
-    create_time: Optional[datetime] = Field(None, description="训练时间")
+    create_datetime: DatetimeStr
 
     model_config = ConfigDict(from_attributes=True)