Merge pull request kpu#304 from owo/py_arpacomplain

Added arpa_complain option to Config python wrapper.
linkerr · Sep 30, 2020 · ba2502b · ba2502b
2 parents b63a09c + 63537d9
commit ba2502b
Show file tree

Hide file tree

Showing 3 changed files with 904 additions and 277 deletions.
diff --git a/python/_kenlm.pxd b/python/_kenlm.pxd
@@ -39,12 +39,19 @@ cdef extern from "util/mmap.hh" namespace "util":
         READ
         PARALLEL_READ
 
+cdef extern from "lm/config.hh" namespace "lm::ngram::Config":
+    cdef enum ARPALoadComplain:
+        ALL
+        EXPENSIVE
+        NONE
+
 cdef extern from "lm/config.hh" namespace "lm::ngram":
     cdef cppclass Config:
         Config()
         float probing_multiplier
         LoadMethod load_method
         bool show_progress
+        ARPALoadComplain arpa_complain
         float unknown_missing_logprob
 
 cdef extern from "lm/model.hh" namespace "lm::ngram":