Skip to content

Commit

Permalink
fix (#107)
Browse files Browse the repository at this point in the history
Co-authored-by: Maxim Mityutko <[email protected]>
  • Loading branch information
maxim-mityutko and Maxim Mityutko authored Apr 11, 2024
1 parent b07ebfb commit 35d0824
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions brickflow/codegen/databricks_bundle.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)))

0 comments on commit 35d0824

Please sign in to comment.