[params] Now use default value if not in the param file

This commit is contained in:
Noe Brucy
2021-07-24 15:46:48 +02:00
parent 5a970dca91
commit a3318595ef
2 changed files with 11 additions and 4 deletions
+9 -2
View File
@@ -26,11 +26,18 @@ _loader.add_implicit_resolver(
)
def load_params(filename):
def params_from_file(filename):
with open(filename) as f:
para_disk = yaml.load(f, Loader=_loader)
return munch.munchify(para_disk)
def default_params():
return load_params(_dir_path + "/params.yml")
return params_from_file(_dir_path + "/params.yml")
def load_params(filename):
basic_params = default_params()
new_params = params_from_file(filename)
basic_params.update(new_params)
return basic_params