"""Diffulex model package that imports built-in models to trigger registration."""

from __future__ import annotations
import importlib
from pathlib import Path

# Import built-in models so their registrations run at import time.
# Automatically import all Python files except auto_model and __init__
_excluded_modules = {"auto_model", "__init__"}
_model_modules = []

_current_dir = Path(__file__).parent
for py_file in _current_dir.glob("*.py"):
    module_name = py_file.stem
    if module_name not in _excluded_modules:
        try:
            importlib.import_module(f".{module_name}", __name__)
            _model_modules.append(module_name)
        except Exception as e:
            # Skip modules that fail to import
            import warnings

            # ImportWarning is ignored by default, which can hide real registration problems.
            warnings.warn(f"Failed to import {module_name}: {e!r}", RuntimeWarning)

__all__ = _model_modules.copy()

from .auto_model import AutoModelForDiffusionLM