surface_to_key_stem dict actually stores rule object now

jtauber · Apr 26, 2016 · b7d283f · b7d283f
1 parent 3253f1c
commit b7d283f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/inflexion/stemming.py b/inflexion/stemming.py
@@ -14,7 +14,7 @@ def __init__(self):
     def add(self, key, rule):
         r = SandhiRule(rule)
         self.key_to_rules[key].append(r)
-        self.surface_to_key_stem[r.surface].add((key, r.stem))
+        self.surface_to_key_stem[r.surface].add((key, r))
         return r
 
     def inflect(self, stem, key):
@@ -47,5 +47,5 @@ def inflect(self, stem, key):
     def possible_stems(self, form):
         for surface, rules in self.surface_to_key_stem.items():
             if form.endswith(surface):
-                for key, stem_ending in rules:
-                    yield (key, form[:-len(surface)] + stem_ending)
+                for key, rule in rules:
+                    yield (key, form[:-len(surface)] + rule.stem)