Fix loading yaml (#5619)

* fix yaml

* chlog
This commit is contained in:
Jirka Borovec 2021-01-25 16:17:55 +01:00 committed by GitHub
parent 30f31d32c8
commit 304f9c5bca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 2 deletions

View File

@ -119,6 +119,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
- Fixed `num_workers` for Windows example ([#5375](https://github.com/PyTorchLightning/pytorch-lightning/pull/5375))
- Fixed loading yaml ([#5619](https://github.com/PyTorchLightning/pytorch-lightning/pull/5619))
## [1.1.3rc] - 2020-12-29
### Added

View File

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

View File

@ -4,7 +4,7 @@ numpy>=1.16.6
torch>=1.4
future>=0.17.1 # required for builtins in setup.py
# pyyaml>=3.13
PyYAML>=5.1, !=5.4.* # OmegaConf requirement >=5.1
PyYAML>=5.1 # OmegaConf requirement >=5.1
tqdm>=4.41.0
fsspec[http]>=0.8.1
tensorboard>=2.2.0