diff --git a/bytelatent/config_parser.py b/bytelatent/config_parser.py index 2e310a2..4c2e2b3 100644 --- a/bytelatent/config_parser.py +++ b/bytelatent/config_parser.py @@ -58,6 +58,11 @@ def parse_args_with_default( if default_cfg is not None: ordered_cfgs.insert(0, default_cfg) cfg = OmegaConf.merge(*ordered_cfgs) + # TODO: Change sources to list[tuple,str, float]] so that this special case isn't needed + for c in reversed(ordered_cfgs): + if "data" in c and "sources" in c["data"]: + cfg["data"]["sources"] = c["data"]["sources"] + break return OmegaConf.to_container(cfg, resolve=True, throw_on_missing=True)