fixup! fix(providers/openlineage): fix common.compat provider ImportE…

…rror handling
apache · Nov 16, 2024 · 7cf91bd · 7cf91bd
1 parent 96d6a58
commit 7cf91bd
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/providers/src/airflow/providers/openlineage/utils/utils.py b/providers/src/airflow/providers/openlineage/utils/utils.py
@@ -76,9 +76,14 @@
     try:
         from airflow.providers.common.compat.assets import Asset
     except ImportError:
-        try:
-            from airflow.providers.common.compat.assets import Asset
-        except ModuleNotFoundError:
+        from packaging.version import Version
+
+        from airflow import __version__ as AIRFLOW_VERSION
+
+        AIRFLOW_V_3_0_PLUS = Version(Version(AIRFLOW_VERSION).base_version) >= Version("3.0.0")
+        if AIRFLOW_V_3_0_PLUS:
+            from airflow.sdk.definitions.asset import Asset
+        else:
             # dataset is renamed to asset since Airflow 3.0
             from airflow.datasets import Dataset as Asset