Remove condition import for SQLAlcRow

astronomer · May 2, 2024 · ea641af · ea641af
1 parent 677151d
commit ea641af
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/python-sdk/src/astro/sql/operators/raw_sql.py b/python-sdk/src/astro/sql/operators/raw_sql.py
@@ -9,15 +9,11 @@
 except ImportError:  # pragma: no cover
     from airflow.decorators import _TaskDecorator as TaskDecorator  # type: ignore[attr-defined]
 
-import airflow
 import pandas as pd
 from airflow.decorators.base import task_decorator_factory
 from sqlalchemy.engine import ResultProxy
 
-if airflow.__version__ >= "2.3":
-    from sqlalchemy.engine.row import LegacyRow as SQLAlcRow
-else:
-    from sqlalchemy.engine.result import RowProxy as SQLAlcRow
+from sqlalchemy.engine.row import LegacyRow as SQLAlcRow
 
 from astro import settings
 from astro.constants import RunRawSQLResultFormat