Skip to content

Commit 03a2fce

Browse files
committed
fix: Also use TI for xcom_push in Airflow 3.0
1 parent ae9ff67 commit 03a2fce

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • airflow_dbt_python/operators

airflow_dbt_python/operators/dbt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from airflow.providers.common.compat.sdk import BaseOperator
1616

1717
from airflow_dbt_python.utils.enums import LogFormat
18-
from airflow_dbt_python.utils.version import AIRFLOW_V_3_1_PLUS
18+
from airflow_dbt_python.utils.version import AIRFLOW_V_3_0_PLUS
1919

2020
if TYPE_CHECKING:
2121
from dbt.contracts.results import RunResult
@@ -346,7 +346,7 @@ def xcom_push_dbt_results(self, context, dbt_results: DbtTaskResult) -> None:
346346
serializable_result = self.make_run_results_serializable(
347347
dbt_results.run_results
348348
)
349-
if AIRFLOW_V_3_1_PLUS:
349+
if AIRFLOW_V_3_0_PLUS:
350350
xcom_push = context["ti"].xcom_push
351351
else:
352352
xcom_push = functools.partial(self.xcom_push, context) # type: ignore

0 commit comments

Comments
 (0)