reference/code/Diffulex-mbd-lms/diffulex/__init__.py

"""Diffulex package root.

Keep this module lightweight so that importing does not eagerly
import the full engine/kernel.
"""

from __future__ import annotations

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    # These are available for type checkers; runtime import is lazy via __getattr__.
    from diffulex.diffulex import Diffulex as Diffulex  # noqa: F401
    from diffulex.sampling_params import SamplingParams as SamplingParams  # noqa: F401
    from diffulex.logger import (
        get_logger as get_logger,
        setup_logger as setup_logger,
        LoggerMixin as LoggerMixin,
    )  # noqa: F401


def __getattr__(name: str):
    if name == "Diffulex":
        # Only trigger heavy side-effect imports when users actually construct the engine.
        # Keeps import lightweight.
        from diffulex import strategy as _strategy  # noqa: F401
        from diffulex.diffulex import Diffulex

        return Diffulex

    if name == "SamplingParams":
        from diffulex.sampling_params import SamplingParams

        return SamplingParams

    if name == "get_logger":
        from diffulex.logger import get_logger

        return get_logger

    if name == "setup_logger":
        from diffulex.logger import setup_logger

        return setup_logger

    if name == "LoggerMixin":
        from diffulex.logger import LoggerMixin

        return LoggerMixin

    raise AttributeError(name)


__all__ = ["Diffulex", "SamplingParams", "get_logger", "setup_logger", "LoggerMixin"]