diff --git a/spacy/kb/__init__.py b/spacy/kb/__init__.py index 3ce3e4c33..93a65ab61 100644 --- a/spacy/kb/__init__.py +++ b/spacy/kb/__init__.py @@ -1,3 +1,11 @@ from .candidate import Candidate, get_candidates, get_candidates_batch from .kb import KnowledgeBase from .kb_in_memory import InMemoryLookupKB + +__all__ = [ + "Candidate", + "KnowledgeBase", + "InMemoryLookupKB", + "get_candidates", + "get_candidates_batch", +] diff --git a/spacy/matcher/__init__.py b/spacy/matcher/__init__.py index f671f2e35..b6d6d70ab 100644 --- a/spacy/matcher/__init__.py +++ b/spacy/matcher/__init__.py @@ -3,4 +3,4 @@ from .levenshtein import levenshtein from .matcher import Matcher from .phrasematcher import PhraseMatcher -__all__ = ["Matcher", "PhraseMatcher", "DependencyMatcher", "levenshtein"] +__all__ = ["DependencyMatcher", "Matcher", "PhraseMatcher", "levenshtein"] diff --git a/spacy/pipeline/__init__.py b/spacy/pipeline/__init__.py index 40e3fd638..2c4a5a8a8 100644 --- a/spacy/pipeline/__init__.py +++ b/spacy/pipeline/__init__.py @@ -22,6 +22,7 @@ from .trainable_pipe import TrainablePipe __all__ = [ "AttributeRuler", "DependencyParser", + "EditTreeLemmatizer", "EntityLinker", "EntityRecognizer", "EntityRuler", diff --git a/spacy/tokens/__init__.py b/spacy/tokens/__init__.py index f4b2bf022..3393ca6ec 100644 --- a/spacy/tokens/__init__.py +++ b/spacy/tokens/__init__.py @@ -5,4 +5,4 @@ from .span import Span from .span_group import SpanGroup from .token import Token -__all__ = ["Doc", "Token", "Span", "SpanGroup", "DocBin", "MorphAnalysis"] +__all__ = ["Doc", "DocBin", "MorphAnalysis", "Span", "SpanGroup", "Token"] diff --git a/spacy/training/__init__.py b/spacy/training/__init__.py index b8c0792f0..5c2ba9932 100644 --- a/spacy/training/__init__.py +++ b/spacy/training/__init__.py @@ -16,3 +16,28 @@ from .iob_utils import ( # noqa: F401 tags_to_entities, ) from .loggers import console_logger # noqa: F401 + +__all__ = [ + "Alignment", + "Corpus", + "Example", + "JsonlCorpus", + "PlainTextCorpus", + "biluo_tags_to_offsets", + "biluo_tags_to_spans", + "biluo_to_iob", + "create_copy_from_base_model", + "docs_to_json", + "dont_augment", + "iob_to_biluo", + "minibatch_by_padded_size", + "minibatch_by_words", + "offsets_to_biluo_tags", + "orth_variants_augmenter", + "read_json_file", + "remove_bilu_prefix", + "split_bilu_label", + "tags_to_entities", + "validate_get_examples", + "validate_examples", +]