fix a bug in config

This commit is contained in:
chenfeiyu 2020-11-23 13:23:13 +08:00
parent 2ed26d3416
commit 598d813908
1 changed files with 3 additions and 3 deletions

View File

@ -21,17 +21,17 @@ class Config(attrdict.AttrDict):
def merge_file(self, path):
with open(path, 'rt') as f:
other = yaml.safe_load(f)
self.update(other)
self.update(self + other)
def merge_args(self, args):
args_dict = vars(args)
args_dict.pop("config") # exclude config file path
args_dict = {k: v for k, v in args_dict.items() if v is not None}
nested_dict = flatdict.FlatDict(args_dict, delimiter=".").as_dict()
self.update(nested_dict)
self.update(self + nested_dict)
def merge(self, other):
self.update(other)
self.update(self + other)
def flatten(self):
flat = flatdict.FlatDict(self, delimiter='.')