Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use full yaml loader (load is deprecated) (#3357)

This commit is contained in:
Sergiu Danalachi 2020-09-07 10:00:35 +02:00 committed by GitHub
parent 69e3f904df
commit b4256df121
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/core/saving.py
View File

@ -337,7 +337,7 @@ def load_hparams_from_yaml(config_yaml: str) -> Dict[str, Any]:
return {} return {}
with fs.open(config_yaml, "r") as fp: with fs.open(config_yaml, "r") as fp:
tags = yaml.load(fp) tags = yaml.full_load(fp)
return tags return tags