yaml.load -> yaml.safe_load

9ed44c85 · PM · Kentaro Wada · 5daf8689 · 9ed44c85
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

labelme/main.py labelme/main.py +2 -2

未找到文件。
--- a/labelme/main.py
+++ b/labelme/main.py
@@ -130,9 +130,9 @@ def main():
    if hasattr(args, 'label_flags'):
        if os.path.isfile(args.label_flags):
            with codecs.open(args.label_flags, 'r', encoding='utf-8') as f:
-                args.label_flags = yaml.load(f)
+                args.label_flags = yaml.safe_load(f)
        else:
-            args.label_flags = yaml.load(args.label_flags)
+            args.label_flags = yaml.safe_load(args.label_flags)

    config_from_args = args.__dict__
    config_from_args.pop('version')