diff --git a/check_ie_result/check_ie_result.py b/check_ie_result/check_ie_result.py index 208f1ce..065f7c9 100644 --- a/check_ie_result/check_ie_result.py +++ b/check_ie_result/check_ie_result.py @@ -52,7 +52,7 @@ def check_ie_result(pk_phhd): discharge = session.query(ZxIeDischarge.pk_ie_discharge, ZxIeDischarge.hospital, ZxIeDischarge.pk_yljg, ZxIeDischarge.department, ZxIeDischarge.pk_ylks, ZxIeDischarge.name, ZxIeDischarge.admission_date, ZxIeDischarge.discharge_date, - ZxIeDischarge.doctor, ZxIeDischarge.admission_id).filter( + ZxIeDischarge.doctor, ZxIeDischarge.admission_id, ZxIeDischarge.age).filter( ZxIeDischarge.pk_phhd == pk_phhd).one() discharge_result = { "pk_ie_discharge": discharge.pk_ie_discharge, @@ -61,6 +61,7 @@ def check_ie_result(pk_phhd): "department": discharge.department, "pk_ylks": discharge.pk_ylks, "name": discharge.name, + "age": discharge.age, "admission_date": discharge.admission_date, "discharge_date": discharge.discharge_date, "doctor": discharge.doctor, diff --git a/visual_model_test/visual_model_test.py b/visual_model_test/visual_model_test.py index 2eb0f95..913f636 100644 --- a/visual_model_test/visual_model_test.py +++ b/visual_model_test/visual_model_test.py @@ -123,7 +123,7 @@ def main(model_type, pic_name=None): elif model_type == "discharge": task_path = "../config/model/discharge_record_model" test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240401000003_3_001938_2.jpg" - schema = ["医院", "科室", "患者姓名", "入院日期", "出院日期", "主治医生", "住院号"] + schema = ["医院", "科室", "患者姓名", "入院日期", "出院日期", "主治医生", "住院号", "年龄"] elif model_type == "cost": task_path = "../config/model/cost_list_model" test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240511000648_4_094542_2.jpg"