From ea641af7247c6a4812cd63aa08a586e2d7943517 Mon Sep 17 00:00:00 2001 From: Pankaj Date: Thu, 2 May 2024 20:24:59 +0530 Subject: [PATCH] Remove condition import for SQLAlcRow --- python-sdk/src/astro/sql/operators/raw_sql.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/python-sdk/src/astro/sql/operators/raw_sql.py b/python-sdk/src/astro/sql/operators/raw_sql.py index 48b9dc21a..8e7d16d3e 100644 --- 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