diff --git a/requirements.txt b/requirements.txt index 56cc61764..c80ae7245 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,8 +2,8 @@ matplotlib>=3.5.2 numpy>=1.24.4 Pillow>=9.0.0 progress>=1.6 -ruamel.yaml>=0.17.21 -ruamel.yaml.clib>=0.2.6 +ruamel.yaml>=0.18.0 +ruamel.yaml.clib>=0.2.7 scikit-image>=0.19.3 scikit-learn>=1.0.2 scipy>=1.10.0 diff --git a/semilearn/core/utils/misc.py b/semilearn/core/utils/misc.py index 2fd183932..dd0dea6da 100644 --- a/semilearn/core/utils/misc.py +++ b/semilearn/core/utils/misc.py @@ -4,7 +4,7 @@ import os import torch import torch.nn as nn -import ruamel.yaml as yaml +from ruamel.yaml import YAML from torch.utils.tensorboard import SummaryWriter def over_write_args_from_dict(args, dict): @@ -21,8 +21,9 @@ def over_write_args_from_file(args, yml): """ if yml == '': return + yaml = YAML(typ='rt') with open(yml, 'r', encoding='utf-8') as f: - dic = yaml.load(f.read(), Loader=yaml.Loader) + dic = yaml.load(f) for k in dic: setattr(args, k, dic[k])