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

2020-09-07 10:00:35 +02:00 · 2020-09-07 10:00:35 +02:00 · b4256df121
parent 69e3f904df
commit b4256df121
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/core/saving.py
+++ b/pytorch_lightning/core/saving.py
@ -337,7 +337,7 @@ def load_hparams_from_yaml(config_yaml: str) -> Dict[str, Any]:
        return {}

    with fs.open(config_yaml, "r") as fp:
-        tags = yaml.load(fp)
+        tags = yaml.full_load(fp)

    return tags