new version

4611a91b · Tomasz Walkowiak · 99c7e165 · 4611a91b
Commit 4611a91b authored Feb 22, 2021 by Tomasz Walkowiak
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/category.py src/category.py +2 -1

No files found.
--- a/src/category.py
+++ b/src/category.py
@@ -43,7 +43,8 @@ class Model:
                self.synsets[str(synset)] = (tag, lemma)
            else:
                if var is not None:
-                    self.variants[lemma.replace(" ","_") + "." + str(var)] = (tag, lemma)
+                    t = lemma.replace(" ", "_") + "." + str(var)
+                    self.variants[t] = (tag, lemma)
                else:
                    self.lemmas[lemma] = tag
        wb.close()