Use CDumper whenever possible (#1152)

Use CDumper whenever possible0
omry · Jan 17, 2024 · d888120 · d888120
1 parent e374b4a
commit d888120
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/news/1152.feature b/news/1152.feature
@@ -0,0 +1 @@
+The YAML dumper will now use `yaml.CDumper` instead of `yaml.Dumper` whenever possible to speed up dumping
diff --git a/omegaconf/_utils.py b/omegaconf/_utils.py
@@ -41,6 +41,13 @@
 except ImportError:  # pragma: no cover
     BaseLoader = yaml.SafeLoader
 
+try:
+    from yaml import CDumper
+
+    BaseDumper = CDumper
+except ImportError:  # pragma: no cover
+    BaseDumper = yaml.Dumper
+
 NoneType: Type[None] = type(None)
 
 BUILTIN_VALUE_TYPES: Tuple[Type[Any], ...] = (
@@ -105,7 +112,7 @@ def __repr__(self) -> str:
 _DEFAULT_MARKER_: Any = Marker("_DEFAULT_MARKER_")
 
 
-class OmegaConfDumper(yaml.Dumper):  # type: ignore
+class OmegaConfDumper(BaseDumper):  # type: ignore
     str_representer_added = False
 
     @staticmethod