From ea641af7247c6a4812cd63aa08a586e2d7943517 Mon Sep 17 00:00:00 2001 From: Pankaj Date: Thu, 2 May 2024 20:24:59 +0530 Subject: [PATCH 1/2] 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 From cb2c1cfd12fb760e75a6c1bdc06b883f47d00361 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 2 May 2024 14:57:03 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- python-sdk/src/astro/sql/operators/raw_sql.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python-sdk/src/astro/sql/operators/raw_sql.py b/python-sdk/src/astro/sql/operators/raw_sql.py index 8e7d16d3e..5c273e5a2 100644 --- a/python-sdk/src/astro/sql/operators/raw_sql.py +++ b/python-sdk/src/astro/sql/operators/raw_sql.py @@ -12,7 +12,6 @@ import pandas as pd from airflow.decorators.base import task_decorator_factory from sqlalchemy.engine import ResultProxy - from sqlalchemy.engine.row import LegacyRow as SQLAlcRow from astro import settings