Spaces:

ajitrajasekharan
/

self-supervised-ner-PER-ORG-LOC

Runtime error

ajitrajasekharan commited on Feb 5, 2022

Commit

846afc5

1 Parent(s): 1cc4362

Update aggregate_server_json.py

Files changed (1) hide show

aggregate_server_json.py CHANGED Viewed

@@ -222,7 +222,7 @@ class AggregateNER:
     def check_if_entity_in_arr(self,entity,arr):
         for node in arr:
-            if (entity == node["e"]):
                 return True
         return False

     def check_if_entity_in_arr(self,entity,arr):
         for node in arr:
+            if (entity == node["e"].split('[')[0]):
                 return True
         return False