From eff7d95c4e9ca980b6b916584c3de4634e4bff2c Mon Sep 17 00:00:00 2001 From: Pavan Kumar Date: Fri, 8 Nov 2024 10:25:54 +0000 Subject: [PATCH] disable aip 44 for lessthan airlfow 3 version --- tests_common/test_utils/compat.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests_common/test_utils/compat.py b/tests_common/test_utils/compat.py index 714cfd734577d..277b4f9c1012b 100644 --- a/tests_common/test_utils/compat.py +++ b/tests_common/test_utils/compat.py @@ -18,6 +18,7 @@ import contextlib import json +import os from typing import TYPE_CHECKING, Any, cast from packaging.version import Version @@ -43,6 +44,11 @@ AIRFLOW_V_2_10_PLUS = Version(AIRFLOW_VERSION.base_version) >= Version("2.10.0") AIRFLOW_V_3_0_PLUS = Version(AIRFLOW_VERSION.base_version) >= Version("3.0.0") +if AIRFLOW_V_3_0_PLUS: + os.environ["AIRFLOW_ENABLE_AIP_44"] = os.environ.get("AIRFLOW_ENABLE_AIP_44", "true") +else: + os.environ["AIRFLOW_ENABLE_AIP_44"] = "false" + try: from airflow.models.baseoperatorlink import BaseOperatorLink except ImportError: