File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
dags/enrich/tasks/airflow_logic Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 7
7
from airflow .operators .bash import BashOperator
8
8
from airflow .operators .python import PythonOperator
9
9
from enrich .config import TASKS , XCOMS , xcom_pull
10
+ from enrich .config .models import EnrichBaseConfig
10
11
11
12
logger = logging .getLogger (__name__ )
12
13
@@ -27,7 +28,7 @@ def task_info_get(dbt_model_refresh_command: str):
27
28
def enrich_dbt_models_refresh_wrapper (ti ) -> None :
28
29
29
30
# Config
30
- config = xcom_pull (ti , XCOMS .CONFIG )
31
+ config : EnrichBaseConfig = xcom_pull (ti , XCOMS .CONFIG )
31
32
32
33
logger .info (task_info_get (config .dbt_models_refresh_command ))
33
34
logger .info (f"📖 Configuration:\n { config .model_dump_json (indent = 2 )} " )
@@ -40,7 +41,7 @@ def enrich_dbt_models_refresh_wrapper(ti) -> None:
40
41
)
41
42
bash = BashOperator (
42
43
task_id = TASKS .ENRICH_DBT_MODELS_REFRESH + "_bash" ,
43
- bash_command = config .dbt_build_command ,
44
+ bash_command = config .dbt_models_refresh_command ,
44
45
)
45
46
bash .execute (context = ti .get_template_context ())
46
47
You can’t perform that action at this time.
0 commit comments