spaCy/spacy/kb/__init__.py

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",
]