Fix

astronomer · Jul 23, 2024 · cc16402 · cc16402
1 parent 39b2f74
commit cc16402
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dev/dags/jaffle_shop_kubernetes.py b/dev/dags/jaffle_shop_kubernetes.py
@@ -9,6 +9,7 @@
 https://astronomer.github.io/astronomer-cosmos/getting_started/kubernetes.html#kubernetes
 
 """
+import os
 from pathlib import Path
 
 from airflow import DAG
@@ -25,8 +26,9 @@
 )
 from cosmos.profiles import PostgresUserPasswordProfileMapping
 
-
-PROJECT_DIR = Path("dags/dbt/jaffle_shop/")
+DEFAULT_DBT_ROOT_PATH = Path(__file__).parent / "dbt"
+DBT_ROOT_PATH = Path(os.getenv("DBT_ROOT_PATH", DEFAULT_DBT_ROOT_PATH))
+PROJECT_DIR = DBT_ROOT_PATH / "jaffle_shop"
 DBT_IMAGE = "dbt-jaffle-shop:1.0.0"
 
 project_seeds = [