type hint

chicago-justice-project · Jan 15, 2020 · f79018e · f79018e
1 parent 7d38545
commit f79018e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/tagnews/senteval/eval.py b/lib/tagnews/senteval/eval.py
@@ -63,7 +63,7 @@ def is_police_entity(self, entity):
                 return entity
             return False
 
-    def extract_google_priority_bin(self, article, cpd_model_val=1, cpd_val=1):
+    def extract_google_priority_bin(self, article:str, cpd_model_val=1, cpd_val=1):
         cop_word_counts = sum([article.count(substr) for substr in self.police_words])
         score = 0.5 * cpd_val + 0.25 * cpd_model_val + 0.25 * min(cop_word_counts / (2 * len(self.police_words)), 1.)
         bin = [bin for bin, bin_max_val in enumerate(bins) if bin_max_val > score][0]