"""DC-QMIX training wrapper."""
from agents.dcqmix_agent import DCQMIXAgent
from training.train_value_based import train_sumo_value_based


def train_sumo_dcqmix(log_dir=None, checkpoint_dir=None, run_timestamp=None):
    return train_sumo_value_based(
        "dcqmix",
        "DC-QMIX",
        DCQMIXAgent,
        log_dir=log_dir,
        checkpoint_dir=checkpoint_dir,
        run_timestamp=run_timestamp,
    )