完善相关问题,并增加推理的部分代码

This commit is contained in:
2025-02-28 16:30:11 +08:00
parent 0301e41e96
commit 4262d3e908
20 changed files with 564 additions and 32 deletions

View File

@ -26,3 +26,14 @@ def get_train_list(project_id: int, session: Session):
query = session.query(ProjectTrain).filter_by(project_id=project_id).order_by(asc(ProjectTrain.id))
train_list = [ProjectTrainOut.from_orm(train) for train in query.all()]
return train_list
def get_train(train_id: int, session: Session):
"""
根据id查询训练信息
:param train_id:
:param session:
:return:
"""
train = session.query(ProjectTrain).filter_by(id=train_id).first()
return train