diff options
| author | tlatorre <tlatorre@uchicago.edu> | 2019-06-05 15:41:20 -0400 |
|---|---|---|
| committer | tlatorre <tlatorre@uchicago.edu> | 2019-06-05 15:41:20 -0400 |
| commit | 94f138d68338e49a94a1518ea4b554ea37d16099 (patch) | |
| tree | 20d0514d4f7d6b96ca7e46f699024980f053ee03 /utils/plot-fit-results | |
| parent | 74c64892d70c2da37954653be7a85dc9efc71f1a (diff) | |
| download | sddm-94f138d68338e49a94a1518ea4b554ea37d16099.tar.gz sddm-94f138d68338e49a94a1518ea4b554ea37d16099.tar.bz2 sddm-94f138d68338e49a94a1518ea4b554ea37d16099.zip | |
try to import CLoader if possible since it's *much* faster
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 |