From 94f138d68338e49a94a1518ea4b554ea37d16099 Mon Sep 17 00:00:00 2001 From: tlatorre Date: Wed, 5 Jun 2019 15:41:20 -0400 Subject: try to import CLoader if possible since it's *much* faster --- utils/plot-fit-results | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'utils/plot-fit-results') 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 -- cgit