Use libYAML's CSafeDumper and CSafeLoader, for speed

2 files changed, 2 insertions, 2 deletions

diff --git a/obbenchlib/htmlgen.py b/obbenchlib/htmlgen.py
index c0bc06f..778be9c 100644
--- a/obbenchlib/htmlgen.py
+++ b/obbenchlib/htmlgen.py
@@ -74,7 +74,7 @@ class HtmlGenerator(object):
         for i, filename in enumerate(filenames):
             print 'Loading', filename, i+1, 'of', len(filenames)
             with open(filename) as f:
-                yield yaml.safe_load(f)
+                yield yaml.load(f, Loader=yaml.CSafeLoader)
 
     def write_file(self, relative_path, text):
         filename = os.path.join(self.htmldir, relative_path)
diff --git a/obbenchlib/result.py b/obbenchlib/result.py
index 8ac24b8..d782680 100644
--- a/obbenchlib/result.py
+++ b/obbenchlib/result.py
@@ -54,4 +54,4 @@ class Result(object):
         }
         filename = os.path.join(dirname, '{}.yaml'.format(self._result_id))
         with open(filename, 'w') as f:
-            yaml.safe_dump(o, stream=f)
+            yaml.dump(o, stream=f, Dumper=yaml.CSafeDumper)