import logging

from model2vec.utils import get_package_extras, importable

_REQUIRED_EXTRA = "train"

for extra_dependency in get_package_extras("model2vec", _REQUIRED_EXTRA):
    importable(extra_dependency, _REQUIRED_EXTRA)

from model2vec.train.classifier import StaticModelForClassification
from model2vec.train.similarity import StaticModelForSimilarity
from model2vec.train.utils import TipFilter

__all__ = ["StaticModelForClassification", "StaticModelForSimilarity"]


logging.getLogger("lightning.pytorch.utilities.rank_zero").addFilter(TipFilter())