diff --git a/src/databricks/labs/blueprint/installation.py b/src/databricks/labs/blueprint/installation.py index fdb573f..188bb9a 100644 --- a/src/databricks/labs/blueprint/installation.py +++ b/src/databricks/labs/blueprint/installation.py @@ -379,6 +379,7 @@ def _overwrite_content(self, filename: str, as_dict: JsonObject, type_ref: type) converters: dict[str, Callable[[Any, type], bytes]] = { "json": self._dump_json, "yml": self._dump_yaml, + "yaml": self._dump_yaml, "csv": self._dump_csv, } extension = filename.split(".")[-1] @@ -421,6 +422,7 @@ def _convert_content(cls, filename: str, raw: BinaryIO) -> JsonObject: converters: dict[str, Callable[[BinaryIO], Any]] = { "json": json.load, "yml": cls._load_yaml, + "yaml": cls._load_yaml, "csv": cls._load_csv, } extension = filename.split(".")[-1]