Skip to content

Commit bd9cf97

Browse files
simplifying file, reducing code
1 parent fccbd45 commit bd9cf97

File tree

1 file changed

+4
-17
lines changed

1 file changed

+4
-17
lines changed

card_data/pipelines/definitions.py

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from pathlib import Path
22

33
from dagster import definitions, load_from_defs_folder
4-
from dagster_dbt import DbtCliResource, DbtProject, dbt_assets
54

65
import dagster as dg
76

@@ -11,31 +10,19 @@
1110

1211
@definitions
1312
def defs():
13+
# load_from_defs_folder discovers dbt assets from transform_data.py
1414
folder_defs = load_from_defs_folder(project_root=Path(__file__).parent.parent)
1515
return dg.Definitions.merge(folder_defs, defs_pricing)
1616

17-
dbt_project_directory = Path(__file__).absolute().parent / "poke_cli_dbt"
18-
dbt_project = DbtProject(project_dir=dbt_project_directory)
19-
20-
dbt_resource = DbtCliResource(project_dir=dbt_project)
21-
22-
# Compiles the dbt project & allow Dagster to build an asset graph
23-
dbt_project.prepare_if_dev()
24-
25-
# Yields Dagster events streamed from the dbt CLI
26-
@dbt_assets(manifest=dbt_project.manifest_path)
27-
def dbt_models(context: dg.AssetExecutionContext, dbt: DbtCliResource):
28-
yield from dbt.cli(["build"], context=context).stream()
29-
30-
# Define the pricing pipeline job that materializes both assets
17+
# Define the pricing pipeline job that materializes the assets and downstream dbt model
3118
pricing_pipeline_job = dg.define_asset_job(
3219
name="pricing_pipeline_job",
33-
selection=dg.AssetSelection.assets(build_dataframe, load_pricing_data),
20+
selection=dg.AssetSelection.assets(build_dataframe, load_pricing_data).downstream(include_self=True),
3421
)
3522

3623
price_schedule = dg.ScheduleDefinition(
3724
name="price_schedule",
38-
cron_schedule="10 10 * * *",
25+
cron_schedule="31 21 * * *",
3926
target=pricing_pipeline_job,
4027
execution_timezone="America/Los_Angeles",
4128
)

0 commit comments

Comments
 (0)