Merge pull request #121 from jelu/yaml

YAML load
DNS-OARC · Aug 22, 2024 · 5a87d86 · 5a87d86
2 parents 4057f9f + 377861f
commit 5a87d86
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/dsc_datatool/transformer/labler.py b/dsc_datatool/transformer/labler.py
@@ -44,10 +44,7 @@ def __init__(self, opts):
         if not 'yaml' in opts:
             raise Exception('yaml=file option required')
         f = open(opts.get('yaml'), 'r', encoding=encoding)
-        try:
-            self.label = yaml.full_load(f)
-        except AttributeError:
-            self.label = yaml.load(f)
+        self.label = yaml.safe_load(f)
         f.close()