From 1d3334f1d933aaf4b15381655b8b8a194bed9e83 Mon Sep 17 00:00:00 2001
From: Sen Wu <senwu@cs.stanford.edu>
Date: Sun, 1 Mar 2020 01:31:40 -0800
Subject: [PATCH] update pyyaml

---
 setup.py             | 2 +-
 src/emmental/meta.py | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 967a324..6d7019b 100644
--- a/setup.py
+++ b/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",
diff --git a/src/emmental/meta.py b/src/emmental/meta.py
index cb466a2..0e1e5c1 100644
--- a/src/emmental/meta.py
+++ b/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