diff options
Diffstat (limited to 'utils/plot-fit-results')
| -rwxr-xr-x | utils/plot-fit-results | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/utils/plot-fit-results b/utils/plot-fit-results index e1f7987..773a0dc 100755 --- a/utils/plot-fit-results +++ b/utils/plot-fit-results @@ -16,6 +16,10 @@ from __future__ import print_function, division import yaml +try: + from yaml import CLoader as Loader +except ImportError: + from yaml.loader import SafeLoader as Loader import numpy as np from scipy.stats import iqr from matplotlib.lines import Line2D @@ -81,7 +85,7 @@ if __name__ == '__main__': for filename in args.filenames: print(filename) with open(filename) as f: - data = yaml.load(f.read()) + data = yaml.load(f.read(),Loader=Loader) a = np.ma.empty(len(data['data']), dtype=[('id',np.int), # particle id |