From 5c08b52cfe02bac1aa39cedf1ef8f92ed050b3e9 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Fri, 5 Jul 2024 08:42:20 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=96=B0=E5=85=B3=E9=94=AE?=
 =?UTF-8?q?=E8=AF=8D=EF=BC=9A=E5=B9=B4=E9=BE=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 check_ie_result/check_ie_result.py     | 3 ++-
 visual_model_test/visual_model_test.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

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"