Update code in lesson 5 and 6

dagster-io · Jul 19, 2024 · 68dbc27 · 68dbc27
1 parent 450161a
commit 68dbc27
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 15 deletions.
diff --git a/...versity/pages/dagster-dbt/lesson-5/2-connecting-dbt-models-to-dagster-assets.md b/...versity/pages/dagster-dbt/lesson-5/2-connecting-dbt-models-to-dagster-assets.md
@@ -105,11 +105,15 @@ Open the `assets/dbt.py` file and do the following:
 At this point, your `dbt.py` file should match the following:
 
 ```python
-import os
+from pathlib import Path
+
 from dagster import AssetExecutionContext, AssetKey
 from dagster_dbt import DagsterDbtTranslator, DbtCliResource, DbtProject, dbt_assets
 
-from .constants import DBT_DIRECTORY
+dbt_project = DbtProject(
+   project_dir=Path(__file__).joinpath("..", "..", "..", "analytics").resolve(),
+)
+dbt_project.prepare_if_dev()
 
 
 class CustomizedDagsterDbtTranslator(DagsterDbtTranslator):
@@ -121,13 +125,10 @@ class CustomizedDagsterDbtTranslator(DagsterDbtTranslator):
         else:
             return super().get_asset_key(dbt_resource_props)
 
-
-dbt_project = DbtProject(project_dir=DBT_DIRECTORY)
-dbt_project.prepare_if_dev()
-
-
+
 @dbt_assets(
-    manifest=dbt_project.manifest_path, dagster_dbt_translator=CustomizedDagsterDbtTranslator()
+    manifest=dbt_project.manifest_path, 
+    dagster_dbt_translator=CustomizedDagsterDbtTranslator(),
 )
 def dbt_analytics(context: AssetExecutionContext, dbt: DbtCliResource):
     yield from dbt.cli(["build"], context=context).stream()

diff --git a/...ter-university/pages/dagster-dbt/lesson-6/3-creating-a-partitioned-dbt-asset.md b/...ter-university/pages/dagster-dbt/lesson-6/3-creating-a-partitioned-dbt-asset.md
@@ -121,17 +121,21 @@ def dbt_analytics(context: AssetExecutionContext, dbt: DbtCliResource):
 At this point, the `dagster_university/assets/dbt.py` file should look like this:
 
 ```python
-import os
 import json
+from pathlib import Path
+
 from dagster import AssetExecutionContext, AssetKey
 from dagster_dbt import DagsterDbtTranslator, DbtCliResource, DbtProject, dbt_assets
 
-from .constants import DBT_DIRECTORY
 from ..partitions import daily_partition
 
-
 INCREMENTAL_SELECTOR = "config.materialized:incremental"
 
+dbt_project = DbtProject(
+   project_dir=Path(__file__).joinpath("..", "..", "..", "analytics").resolve(),
+)
+dbt_project.prepare_if_dev()
+
 
 class CustomizedDagsterDbtTranslator(DagsterDbtTranslator):
     def get_asset_key(self, dbt_resource_props):
@@ -143,10 +147,6 @@ class CustomizedDagsterDbtTranslator(DagsterDbtTranslator):
             return super().get_asset_key(dbt_resource_props)
 
 
-dbt_project = DbtProject(project_dir=DBT_DIRECTORY)
-dbt_project.prepare_if_dev()
-
-
 @dbt_assets(
     manifest=dbt_project.manifest_path,
     dagster_dbt_translator=CustomizedDagsterDbtTranslator(),