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: