Update lama_permutation_stats.py. yaml.load changed to yaml.safe_load

VCCRI · Feb 18, 2024 · b5eb048 · b5eb048
1 parent c618e81
commit b5eb048
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lama/scripts/lama_permutation_stats.py b/lama/scripts/lama_permutation_stats.py
@@ -56,7 +56,7 @@ def p(path):
         return resolved
 
     with open(cfg_path, 'r') as fh:
-        cfg = yaml.load(fh)
+        cfg = yaml.safe_load(fh)
 
     for key in cfg.keys():
         if key not in allowed_cfg_keys: