src/__init__.py

"""Chess Challenge source module."""

from .model import ChessConfig, ChessForCausalLM
from .tokenizer import ChessTokenizer

# Lazy import for evaluate to avoid RuntimeWarning when running as module
def __getattr__(name):
    if name == "ChessEvaluator":
        from .evaluate import ChessEvaluator
        return ChessEvaluator
    if name == "load_model_from_hub":
        from .evaluate import load_model_from_hub
        return load_model_from_hub
    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

__all__ = [
    "ChessConfig",
    "ChessForCausalLM", 
    "ChessTokenizer",
    "ChessEvaluator",
    "load_model_from_hub",
]