fixed bug where scored analyses for MLE disambiguation are not stored…

camel_tools/disambig/mle.py

@@ -225,7 +225,13 @@ def _scored_analyses(self, word_dd):
             return scored_analyses[0:self._top]
 
     def _scored_analyses_cached(self, word_dd):
-        return self._cache.get(word_dd, self._scored_analyses(word_dd))
+        if word_dd in self._cache:
+            scored_analyses = self._cache[word_dd]
+        else:
+            scored_analyses = self._scored_analyses(word_dd)
+            self._cache[word_dd] = scored_analyses
+
+        return scored_analyses
 
     def _disambiguate_word(self, word):
         word_dd = dediac_ar(word)

camel_tools/disambig/test.py

@@ -0,0 +1,17 @@
+from cachetools import LFUCache, cached
+import time
+
+class TestClass:
+    def __init__(self):
+        self._cache = LFUCache(100000)
+
+    @cached(cache=lambda self: self._cache, key=lambda self, word_dd: word_dd)
+    def test_cached(self, word_dd):
+        return word_dd
+
+
+# Example usage:
+test = TestClass()
+test.test_cached("hello")
+
+test.test_cached("hello")