import os

from utils import load_yaml

def get_coder_cfg(cfg, dsf1):
    conv_dir = cfg.cur_dir + '/models/conv/config'
    assert os.path.exists(conv_dir), conv_dir
    
    cfg_path = conv_dir + f'/{dsf1}.yaml'

    ds_cfg = load_yaml(cfg_path)

    enc_cfg = ds_cfg.encoder
    enc2_cfg = ds_cfg.encoder2
    dec_cfg = ds_cfg.decoder
    dec2_cfg = ds_cfg.decoder2
    return enc_cfg, enc2_cfg, dec_cfg, dec2_cfg