update pyyaml

setup.py

@@ -16,7 +16,7 @@
     py_modules=[splitext(basename(path))[0] for path in glob("src/*.py")],
     install_requires=[
         "numpy>=1.11, <2.0",
-        "pyyaml>=4.2b1, <5.0",
+        "pyyaml>=5.1, <6.0",
         "scikit-learn>=0.20.3, <0.21.0",
         "scipy>=1.1.0, <2.0.0",
         "tensorboard>=2.0.1, <3.0.0",

src/emmental/meta.py

@@ -56,7 +56,7 @@ def init_config() -> None:
         os.path.dirname(os.path.realpath(__file__)), "emmental-default-config.yaml"
     )
     with open(default_config_path, "r") as f:
-        config = yaml.load(f)
+        config = yaml.load(f, Loader=yaml.FullLoader)
     logger.info(f"Loading Emmental default config from {default_config_path}.")
 
     Meta.config = config
@@ -167,7 +167,9 @@ def update_config(
                 if os.path.exists(potential_path):
                     with open(potential_path, "r") as f:
                         Meta.config = merge(
-                            Meta.config, yaml.load(f), specical_keys="checkpoint_metric"
+                            Meta.config,
+                            yaml.load(f, Loader=yaml.FullLoader),
+                            specical_keys="checkpoint_metric",
                         )
                     logger.info(f"Updating Emmental config from {potential_path}.")
                     break