disable aip 44 for lessthan airlfow 3 version (apache#43818)

astronomer · Nov 11, 2024 · 41d073e · 41d073e
1 parent f757b87
commit 41d073e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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: