Faster yaml loading using the native loader if available

taskcluster · Mar 21, 2024 · c8d6a7f · c8d6a7f
1 parent d1dabab
commit c8d6a7f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -58,5 +58,6 @@ exclude = [  # TODO fix errors in these files
     "src/taskgraph/util/taskcluster.py",
     "src/taskgraph/util/vcs.py",
     "src/taskgraph/util/workertypes.py",
+    "src/taskgraph/util/yaml.py",
 ]
 reportIncompatibleMethodOverride = false
diff --git a/src/taskgraph/util/yaml.py b/src/taskgraph/util/yaml.py
@@ -4,8 +4,12 @@
 
 
 import os
+import typing
 
-from yaml.loader import SafeLoader
+try:
+    from yaml import CSafeLoader as SafeLoader
+except ImportError:
+    from yaml import SafeLoader
 
 
 class UnicodeLoader(SafeLoader):