Fix key name in context of task

apache · Nov 8, 2024 · 571eaca · 571eaca
1 parent 9240b89
commit 571eaca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/providers/src/airflow/providers/dbt/cloud/operators/dbt.py b/providers/src/airflow/providers/dbt/cloud/operators/dbt.py
@@ -149,7 +149,7 @@ def execute(self, context: Context):
                 self.run_id = non_terminal_runs[0]["id"]
                 job_run_url = non_terminal_runs[0]["href"]
 
-        is_retry = context["task_instance"].try_number != 1
+        is_retry = context["ti"].try_number != 1
 
         if not self.reuse_existing_run or not non_terminal_runs:
             trigger_job_response = self.hook.trigger_job_run(