diff --git a/apps/vinyl/src/definitions.py b/apps/vinyl/src/definitions.py index ad184b9..30a5396 100644 --- a/apps/vinyl/src/definitions.py +++ b/apps/vinyl/src/definitions.py @@ -28,6 +28,8 @@ def deep_merge_dicts(base: dict, override: dict) -> dict: return result +storage_dir = os.environ.get("STORAGE_DIR", "/storage") + definitions = dg.Definitions( assets=[ asset.with_attributes( @@ -37,9 +39,7 @@ definitions = dg.Definitions( for asset in dg.load_assets_from_modules([assets]) ], resources={ - "polars_parquet_io_manager": PolarsParquetIOManager( - base_dir=os.environ.get("STORAGE_DIR", "/storage") - ), + "polars_parquet_io_manager": PolarsParquetIOManager(base_dir=storage_dir), "email_service": EmailService( smtp_server=dg.EnvVar("SMTP_SERVER"), smtp_port=dg.EnvVar.int("SMTP_PORT"),