"""
TD3B: Transition-Directed Discrete Diffusion for Binders
A module extending TR2-D2 with directional allosteric control.
"""

from .direction_oracle import DirectionalOracle
from .td3b_scoring import TD3BRewardFunction, TD3BConfidenceWeighting, create_td3b_reward_function
from .td3b_losses import ContrastiveLoss, InfoNCELoss, TD3BTotalLoss, extract_embeddings_from_mdlm
from .td3b_mcts import TD3B_MCTS, create_td3b_mcts
from .td3b_finetune import td3b_finetune, add_td3b_sampling_to_model
from .data_utils import TD3BDataset, load_td3b_data

__all__ = [
    'DirectionalOracle',
    'TD3BRewardFunction',
    'TD3BConfidenceWeighting',
    'create_td3b_reward_function',
    'ContrastiveLoss',
    'InfoNCELoss',
    'TD3BTotalLoss',
    'extract_embeddings_from_mdlm',
    'TD3B_MCTS',
    'create_td3b_mcts',
    'td3b_finetune',
    'add_td3b_sampling_to_model',
    'TD3BDataset',
    'load_td3b_data',
]

__version__ = '0.1.0'