Srinivasan Iyer sviyer commited on
Commit
a5ceaaa
·
unverified ·
1 Parent(s): 7517ac2

When merging configs, do not merge data sources (#79)

Browse files

* When merging configs, do not merge data sources

* Add todo

---------

Co-authored-by: Srini Iyer <sviyer@meta.com>

Files changed (1) hide show
  1. bytelatent/config_parser.py +5 -0
bytelatent/config_parser.py CHANGED
@@ -58,6 +58,11 @@ def parse_args_with_default(
58
  if default_cfg is not None:
59
  ordered_cfgs.insert(0, default_cfg)
60
  cfg = OmegaConf.merge(*ordered_cfgs)
 
 
 
 
 
61
  return OmegaConf.to_container(cfg, resolve=True, throw_on_missing=True)
62
 
63
 
 
58
  if default_cfg is not None:
59
  ordered_cfgs.insert(0, default_cfg)
60
  cfg = OmegaConf.merge(*ordered_cfgs)
61
+ # TODO: Change sources to list[tuple,str, float]] so that this special case isn't needed
62
+ for c in reversed(ordered_cfgs):
63
+ if "data" in c and "sources" in c["data"]:
64
+ cfg["data"]["sources"] = c["data"]["sources"]
65
+ break
66
  return OmegaConf.to_container(cfg, resolve=True, throw_on_missing=True)
67
 
68