Update project.py

maximearmstrong · Jul 23, 2024 · 4606b2e · 4606b2e
1 parent 70c5bb6
commit 4606b2e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/jaffle_dagster/jaffle_dagster/project.py b/jaffle_dagster/jaffle_dagster/project.py
@@ -1,9 +1,15 @@
 from pathlib import Path
 
 from dagster_dbt import DbtProject
+from dagster_dbt.dbt_project import using_dagster_dev
+
+if using_dagster_dev():
+    project_dir = Path(__file__).joinpath("..", "..", "..", "jaffle_shop").resolve()
+else:
+    project_dir = Path(__file__).joinpath("..", "..", "jaffle_shop").resolve()
 
 jaffle_shop_project = DbtProject(
-    project_dir=Path(__file__).joinpath("..", "..", "..", "jaffle_shop").resolve(),
+    project_dir=project_dir,
     packaged_project_dir=Path(__file__).joinpath("..", "..", "dbt-project").resolve(),
 )
-jaffle_shop_project.prepare_if_dev()
+jaffle_shop_project.prepare_if_dev()