diff --git a/dsc_datatool/transformer/labler.py b/dsc_datatool/transformer/labler.py index e510511..2fc8120 100644 --- 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()