diff --git a/brickflow/codegen/databricks_bundle.py b/brickflow/codegen/databricks_bundle.py index fbd2f2e1..62873d73 100644 --- a/brickflow/codegen/databricks_bundle.py +++ b/brickflow/codegen/databricks_bundle.py @@ -666,5 +666,11 @@ def synth(self) -> None: if self.env == BrickflowDefaultEnvs.LOCAL.value: # it should use the bundles project env field as that is the folder bundles makes ImportManager.create_import_tf(get_bundles_project_env(), self.imports) + + def quoted_presenter(dumper, data): # type: ignore[no-untyped-def] + return dumper.represent_scalar("tag:yaml.org,2002:str", data, style='"') + + yaml.add_representer(str, quoted_presenter) + with open("bundle.yml", "w", encoding="utf-8") as f: f.write(yaml.dump(bundle.dict(exclude_unset=True)))