diff --git a/dags/sustainability_score/__init__.py b/dags/sustainability_score/__init__.py index 8842365..f1072a9 100644 --- a/dags/sustainability_score/__init__.py +++ b/dags/sustainability_score/__init__.py @@ -23,8 +23,8 @@ CSV_FNAME = ( CONFIG = { "input": f"{ HOME }/gcs/data/{ CSV_FNAME }", "beam_etl_path": "/etl/main.py", + "dbt_path": "/dbt", "products_table": "sustainability_score.products", - "scored_table": "sustainability_score.scored_products", } @@ -34,7 +34,7 @@ def dbt(cmd: str, attach_dag: DAG) -> BashOperator: dag=attach_dag, task_id=f"dbt_{ cmd }", bash_command=f"dbt { cmd }", - cwd="/dbt", + cwd="{{ params.dbt_path }}", env={ "POSTGRES_HOST": "{{ conn.get('pg_db').host }}", "POSTGRES_USER": "{{ conn.get('pg_db').login }}",