Function smart_transform

DataLoader/SequenceBase.py:120–141

(seq: SequenceBase[T_Data], trans_cfg: SimpleNamespace | dict[str, Any] | list)

def smart_transform(seq: SequenceBase[T_Data], trans_cfg: SimpleNamespace | dict[str, Any] | list) -> SequenceBase[T_Data]:
    if isinstance(trans_cfg, dict):
        trans_cfg = build_dynamic_config(trans_cfg)[0]
    elif isinstance(trans_cfg, list):
        trans_cfg = [
            tcfg if isinstance(tcfg, SimpleNamespace) else build_dynamic_config(tcfg)[0]
            for tcfg in trans_cfg
        ]
    
    if isinstance(trans_cfg, list):
        transform_cfg = trans_cfg
    else:
        seq_type = seq.name()
        if not hasattr(trans_cfg, seq_type): return seq
        transform_cfg = getattr(trans_cfg, seq_type)
    
    Logger.write("info", "Using data transformation:\n- " + '\n- '.join([str(x) for x in transform_cfg]))
    trans_fn = [
        IDataTransform.instantiate(tcfg.type, tcfg.args)
        for tcfg in transform_cfg
    ]
    return seq.transform(trans_fn)