From 5da35efe4b81809d92419c829453557921f9190f Mon Sep 17 00:00:00 2001 From: trylks Date: Tue, 25 Sep 2018 14:24:17 +0200 Subject: [PATCH] Check extensions before setting them Not doing so causes some easily avoidable errors (easily here). This way the code is more robust. --- spacy_hunspell/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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: