Faster yaml loading using the native loader if available

serge-sans-paille · ahal · commit c8d6a7f44f1f · 2024-03-21T11:47:52.000-04:00
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):

Original file line number	Diff line number	Diff line change
`@@ -58,5 +58,6 @@ exclude = [ # TODO fix errors in these files`
`58`	`58`	`"src/taskgraph/util/taskcluster.py",`
`59`	`59`	`"src/taskgraph/util/vcs.py",`
`60`	`60`	`"src/taskgraph/util/workertypes.py",`
	`61`	`+ "src/taskgraph/util/yaml.py",`
`61`	`62`	`]`
`62`	`63`	`reportIncompatibleMethodOverride = false`