setup: use explicit re-exports (#151)

* setup: use explicit re-exports * fix tests
adbar · Oct 9, 2024 · 10cc7fb · 10cc7fb
1 parent ebabe83
commit 10cc7fb
Show file tree

Hide file tree

Showing 5 changed files with 49 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -17,7 +17,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13-dev']
+        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13'] # '3.14-dev'
         env: [{ MINIMAL: "true" }, { MINIMAL: "false" }]
         include:
         # common versions on MacOS

diff --git a/simplemma/__init__.py b/simplemma/__init__.py
@@ -20,3 +20,26 @@
     TokenSampler,
 )
 from .tokenizer import RegexTokenizer, Tokenizer, simple_tokenizer
+
+__all__ = [
+    "__title__",
+    "__author__",
+    "__email__",
+    "__license__",
+    "__version__",
+    "LanguageDetector",
+    "in_target_language",
+    "langdetect",
+    "Lemmatizer",
+    "is_known",
+    "lemma_iterator",
+    "lemmatize",
+    "text_lemmatizer",
+    "BaseTokenSampler",
+    "MostCommonTokenSampler",
+    "RelaxedMostCommonTokenSampler",
+    "TokenSampler",
+    "RegexTokenizer",
+    "Tokenizer",
+    "simple_tokenizer",
+]
diff --git a/simplemma/strategies/__init__.py b/simplemma/strategies/__init__.py
@@ -16,3 +16,20 @@
 from .lemmatization_strategy import LemmatizationStrategy
 from .prefix_decomposition import PrefixDecompositionStrategy
 from .rules import RulesStrategy
+
+__all__ = [
+    "AffixDecompositionStrategy",
+    "DefaultStrategy",
+    "DefaultDictionaryFactory",
+    "DictionaryFactory",
+    "TrieDictionaryFactory",
+    "DictionaryLookupStrategy",
+    "LemmatizationFallbackStrategy",
+    "RaiseErrorFallbackStrategy",
+    "ToLowercaseFallbackStrategy",
+    "GreedyDictionaryLookupStrategy",
+    "HyphenRemovalStrategy",
+    "LemmatizationStrategy",
+    "PrefixDecompositionStrategy",
+    "RulesStrategy",
+]
diff --git a/simplemma/strategies/dictionaries/__init__.py b/simplemma/strategies/dictionaries/__init__.py
@@ -2,3 +2,5 @@
 
 from .dictionary_factory import DefaultDictionaryFactory, DictionaryFactory
 from .trie_directory_factory import TrieDictionaryFactory
+
+__all__ = ["DefaultDictionaryFactory", "DictionaryFactory", "TrieDictionaryFactory"]
diff --git a/simplemma/strategies/fallback/__init__.py b/simplemma/strategies/fallback/__init__.py
@@ -3,3 +3,9 @@
 from .lemmatization_fallback_strategy import LemmatizationFallbackStrategy
 from .raise_error import RaiseErrorFallbackStrategy
 from .to_lowercase import ToLowercaseFallbackStrategy
+
+__all__ = [
+    "LemmatizationFallbackStrategy",
+    "RaiseErrorFallbackStrategy",
+    "ToLowercaseFallbackStrategy",
+]