alpha_lab/cta_1d/src/__init__.py

"""CTA 1-day task-specific utilities."""

from .loader import CTA1DLoader
from .labels import get_blend_weights, describe_blend_config, BLEND_CONFIGS

try:
    from .train import train_model, TrainConfig
    from .backtest import run_backtest, BacktestConfig
    __all__ = [
        'CTA1DLoader',
        'train_model',
        'TrainConfig',
        'run_backtest',
        'BacktestConfig',
        'get_blend_weights',
        'describe_blend_config',
        'BLEND_CONFIGS',
    ]
except ImportError:
    # xgboost or sklearn not installed
    __all__ = [
        'CTA1DLoader',
        'get_blend_weights',
        'describe_blend_config',
        'BLEND_CONFIGS',
    ]