police words in separate file

chicago-justice-project · Nov 6, 2019 · 89287f5 · 89287f5
1 parent c6ae60f
commit 89287f5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/tagnews/senteval/eval.py b/lib/tagnews/senteval/eval.py
@@ -21,6 +21,7 @@ def process_google_result(text):
 class SentimentGoogler:
     def __init__(self):
         self.client = self.connect_to_client()
+        self.polce_words = police_words
 
     def run(self, doc_text):
         sentiment_ = self.call_api(doc_text)
@@ -61,10 +62,10 @@ def is_police_entity(self, entity):
             "pigs",
             "policeofficer",
         ]
-        if entity in possible_responses:
+        if entity in self.police_words:
             return entity
         for mention in entity.mentions:
-            if pre_process_text(mention.text.content) in possible_responses:
+            if pre_process_text(mention.text.content) in self.police_words:
                 return entity
             return False