fix: Declare required token annotation "is_hit" in main module

zentrum-lexikographie · Feb 21, 2025 · af2b93c · af2b93c
1 parent e124586
commit af2b93c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/gdex/__init__.py b/gdex/__init__.py
@@ -13,6 +13,7 @@
 _DEFAULT_RARE_CHARS = "0123456789')(-"
 
 Span.set_extension("gdex", default=0.0)
+Token.set_extension("is_hit", default=False)
 
 
 @dataclass

diff --git a/tests/test_gdex.py b/tests/test_gdex.py
@@ -4,8 +4,6 @@
 
 import gdex
 
-spacy.tokens.Token.set_extension("is_hit", default=False)
-
 spacy_model_packages = {
     "de_core_news_sm": (
         "de-core-news-sm @ https://github.com/explosion/spacy-models/"