#!/usr/bin/env python3
"""Skip Vision-Exp gate tensors the B12X DSpark draft loader cannot bind."""
from pathlib import Path

import vllm

base = Path(vllm.__file__).parent / "models" / "deepseek_v4" / "nvidia"
for name in ("dspark.py", "dspark_v2.py"):
    if (base / name).exists():
        p = base / name
        break
else:
    raise SystemExit("no dspark.py/dspark_v2.py under " + str(base))
text = p.read_text()
old = '''                if name.endswith(".ffn.gate.bias"):
                    name = name.replace(
                        ".ffn.gate.bias", ".ffn.gate.e_score_correction_bias"
                    )
                param = params_dict[name]
'''
new = '''                if name.endswith(".ffn.gate.bias_vl"):
                    continue
                if name.endswith(".ffn.gate.bias"):
                    name = name.replace(
                        ".ffn.gate.bias", ".ffn.gate.e_score_correction_bias"
                    )
                if name not in params_dict:
                    continue
                param = params_dict[name]
'''
if old not in text:
    raise SystemExit(f"patch target missing in {p}")
p.write_text(text.replace(old, new, 1))