replace deprecated yaml.safe_load function

LorenFrankLab · Feb 19, 2024 · 7ecc7d8 · 7ecc7d8
1 parent 4d33438
commit 7ecc7d8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spyglass/position/v1/dlc_reader.py b/src/spyglass/position/v1/dlc_reader.py
@@ -114,7 +114,8 @@ def pkl(self):
     def yml(self):
         if self._yml is None:
             with open(self.yml_path, "rb") as f:
-                self._yml = yaml.safe_load(f)
+                safe_yaml = yaml.YAML(typ="safe", pure=True)
+                self._yml = safe_yaml.load(f)
         return self._yml
 
     @property