diff --git a/qlib/__init__.py b/qlib/__init__.py index 20c2cab4a3..99035e5014 100644 --- a/qlib/__init__.py +++ b/qlib/__init__.py @@ -147,7 +147,7 @@ def init_from_yaml_conf(conf_path, **kwargs): """ with open(conf_path) as f: - config = yaml.load(f, Loader=yaml.SafeLoader) + config = yaml.safe_load(f) config.update(kwargs) default_conf = config.pop("default_conf", "client") init(default_conf, **config) diff --git a/qlib/contrib/online/manager.py b/qlib/contrib/online/manager.py index cf850b9dac..b0be7f0768 100644 --- a/qlib/contrib/online/manager.py +++ b/qlib/contrib/online/manager.py @@ -110,7 +110,7 @@ def add_user(self, user_id, config_file, add_date): raise ValueError("User data for {} already exists".format(user_id)) with config_file.open("r") as fp: - config = yaml.load(fp) + config = yaml.safe_load(fp) # load model model = init_instance_by_config(config["model"]) diff --git a/qlib/contrib/online/utils.py b/qlib/contrib/online/utils.py index 611af63e4a..71a6a91ec2 100644 --- a/qlib/contrib/online/utils.py +++ b/qlib/contrib/online/utils.py @@ -88,7 +88,7 @@ def prepare(um, today, user_id, exchange_config=None): dates.append(get_next_trading_date(dates[-1], future=True)) if exchange_config: with pathlib.Path(exchange_config).open("r") as fp: - exchange_paras = yaml.load(fp) + exchange_paras = yaml.safe_load(fp) else: exchange_paras = {} trade_exchange = Exchange(trade_dates=dates, **exchange_paras) diff --git a/qlib/contrib/tuner/config.py b/qlib/contrib/tuner/config.py index f23d1b8740..247fa6a4fa 100644 --- a/qlib/contrib/tuner/config.py +++ b/qlib/contrib/tuner/config.py @@ -14,7 +14,7 @@ def __init__(self, config_path): self.config_path = config_path with open(config_path) as fp: - config = yaml.load(fp) + config = yaml.safe_load(fp) self.config = copy.deepcopy(config) self.pipeline_ex_config = PipelineExperimentConfig(config.get("experiment", dict()), self) diff --git a/qlib/utils/__init__.py b/qlib/utils/__init__.py index f5e981c24d..be7969b658 100644 --- a/qlib/utils/__init__.py +++ b/qlib/utils/__init__.py @@ -128,10 +128,10 @@ def parse_config(config): # Check whether config is file if os.path.exists(config): with open(config, "r") as f: - return yaml.load(f, Loader=yaml.SafeLoader) + return yaml.safe_load(f) # Check whether the str can be parsed try: - return yaml.load(config) + return yaml.safe_load(config) except BaseException: raise ValueError("cannot parse config!") diff --git a/qlib/workflow/cli.py b/qlib/workflow/cli.py index f7455797bd..6eba962779 100644 --- a/qlib/workflow/cli.py +++ b/qlib/workflow/cli.py @@ -44,7 +44,7 @@ def sys_config(config, config_path): # worflow handler function def workflow(config_path, experiment_name="workflow", uri_folder="mlruns"): with open(config_path) as fp: - config = yaml.load(fp, Loader=yaml.SafeLoader) + config = yaml.safe_load(fp) # config the `sys` section sys_config(config, config_path)