Skip to content

Commit 8524cae

Browse files
irrationalagentfbertsch
authored andcommitted
update forecasting dag
1 parent 6092377 commit 8524cae

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

dags/kpi_forecasting.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,16 @@
6161
dag=dag,
6262
)
6363

64+
kpi_forecasting_mobile_non_cumulative = gke_command(
65+
task_id="kpi_forecasting_mobile_non_cumulative",
66+
command=[
67+
"python", "kpi-forecasting/kpi_forecasting.py",
68+
"-c",
69+
] + ["kpi-forecasting/yaml/mobile_non_cumulative.yaml"],
70+
docker_image="gcr.io/moz-fx-data-airflow-prod-88e0/kpi-forecasting_docker_etl:latest",
71+
dag=dag,
72+
)
73+
6474
wait_for_desktop_usage = ExternalTaskSensor(
6575
task_id="wait_for_desktop_usage",
6676
external_dag_id="bqetl_main_summary",
@@ -104,4 +114,5 @@
104114

105115
wait_for_mobile_usage >> kpi_forecasting_mobile
106116
wait_for_unified_metrics >> kpi_forecasting_desktop
107-
wait_for_desktop_usage >> kpi_forecasting_desktop_non_cumulative
117+
wait_for_unified_metrics >> kpi_forecasting_desktop_non_cumulative
118+
wait_for_unified_metrics >> kpi_forecasting_mobile_non_cumulative

0 commit comments

Comments
 (0)