diff --git a/spacy_hunspell/__init__.py b/spacy_hunspell/__init__.py index 1e6fca3..f4bcf7f 100644 --- a/spacy_hunspell/__init__.py +++ b/spacy_hunspell/__init__.py @@ -33,8 +33,10 @@ def __init__(self, nlp, path=HUNSPELL_PROFILE): self.hobj = HunSpell(dic_path, aff_path) - Token.set_extension('hunspell_spell', default=None) - Token.set_extension('hunspell_suggest', getter=self.get_suggestion) + if not Token.has_extension('hunspell_spell'): + Token.set_extension('hunspell_spell', default=None) + if not Token.has_extension('hunspell_suggest'): + Token.set_extension('hunspell_suggest', getter=self.get_suggestion) def __call__(self, doc): for token in doc: