1 files changed, 6 insertions, 1 deletions
diff --git a/utils/plot b/utils/plot
index 6aa55fa..ac93eda 100755
--- a/utils/plot
+++ b/utils/plot
@@ -16,6 +16,11 @@
 
 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
@@ -77,7 +82,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)
 
         dx = []
         dy = []