diff --git a/teste_dag.py b/teste_dag.py index faeb6ef82646275267455d55b005ed5739a8fb51..04b0585ba49163e46cd0f3da8a9eef1eceb6ead0 100644 --- a/teste_dag.py +++ b/teste_dag.py @@ -31,16 +31,20 @@ with DAG( packages="org.apache.hadoop:hadoop-aws:3.3.4" ) - task_dbt_setup = BashOperator( - task_id="dbt_setup", - bash_command=f"rm -rf /tmp/proj_teste && cp -r {repo_home}/proj_teste /tmp/proj_teste", - ) + @task + def task_dbt_setup(): + BashOperator( + task_id="dbt_setup", + bash_command=f"rm -rf /tmp/proj_teste && cp -r {repo_home}/proj_teste /tmp/proj_teste", + ) - task_bronze_to_gold = BashOperator( - task_id="bronze_to_silver_to_gold", - cwd="/tmp/proj_teste", - bash_command="dbt deps && dbt build", - ) + @task + def task_bronze_to_gold(): + BashOperator( + task_id="bronze_to_silver_to_gold", + cwd="/tmp/proj_teste", + bash_command="dbt deps && dbt build", + ) task_spark >> task_dbt_setup >> task_bronze_to_gold