"""
Ablation: Direct Soft Matching (no prototypes).

Reuses the full training pipeline from train.py, only swapping
CODI_PMPS (prototype-mediated) with CODI_DirectMatch (direct cosine matching).
"""
import train as train_module
from src.model_direct_match import CODI_DirectMatch

# Replace the model class before invoking train()
train_module.CODI_PMPS = CODI_DirectMatch

if __name__ == "__main__":
    train_module.train()