Merge pull request #3 from gokhanalcan/ar

`ar` branch: yaml loader
utiasDSL · Nov 5, 2021 · aed0796 · aed0796
2 parents 76ff24c + 02cf616
commit aed0796
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/safe_control_gym/utils/registration.py b/safe_control_gym/utils/registration.py
@@ -55,7 +55,7 @@ def get_config(self):
                 # Specified as file path.
                 mod_name, config_name = self.config_entry_point.split(":")
                 with pkg_resources.open_text(mod_name, config_name) as f:
-                    config = yaml.load(f)
+                    config = yaml.load(f, Loader=yaml.FullLoader)
             else:
                 # Specified as "module_path:config_dict_name".
                 config = load(self.config_entry_point)

diff --git a/safe_control_gym/utils/utils.py b/safe_control_gym/utils/utils.py
@@ -51,7 +51,7 @@ def read_file(file_path, sep=","):
     if "json" in file_path:
         data = json.load(f)
     elif "yaml" in file_path:
-        data = yaml.load(f)
+        data = yaml.load(f, Loader=yaml.FullLoader)
     else:
         sep = sep if "csv" in file_path else " "
         data = []