瀏覽代碼

修复entity.person_phone为None的报错

znj 3 年之前
父節點
當前提交
51801c3d06
共有 2 個文件被更改,包括 7 次插入6 次删除
  1. 6 5
      BiddingKG/dl/interface/getAttributes.py
  2. 1 1
      BiddingKG/dl/interface/predictor.py

+ 6 - 5
BiddingKG/dl/interface/getAttributes.py

@@ -2265,11 +2265,12 @@ def findAttributeAfterEntity(PackDict,roleSet,PackageList,PackageSet,list_entity
             if PackDict[k]["roleList"][0].role_name == "tenderee":
                 if not PackDict[k]["roleList"][0].linklist:
                     for _entity in temporary_list2[::-1]:
-                        if _entity.entity_type=='person' and _entity.label==3 and len(_entity.person_phone)>0:
-                            _phone = [p.entity_text for p in _entity.person_phone]
-                            for _p in _phone:
-                                PackDict[k]["roleList"][0].linklist.append((_entity.entity_text, _p))
-                            break
+                        if _entity.entity_type=='person' and _entity.label==3:
+                            if _entity.person_phone:
+                                _phone = [p.entity_text for p in _entity.person_phone]
+                                for _p in _phone:
+                                    PackDict[k]["roleList"][0].linklist.append((_entity.entity_text, _p))
+                                break
 
     for pack in PackDict.keys():
         for i in range(len(PackDict[pack]["roleList"])):

+ 1 - 1
BiddingKG/dl/interface/predictor.py

@@ -794,7 +794,7 @@ class EPCPredict():
                 values.append(item)
             # phone_number = phone[i]
             # entity.set_Person(label,values,phone_number)
-            entity.set_Person(label,values,None)
+            entity.set_Person(label,values,[])
         # 为联系人匹配电话
         # self.person_search_phone(list_sentences, list_entitys)