Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Remove condition import for SQLAlcRow #2146

Merged
merged 2 commits into from
May 6, 2024
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 1 addition & 6 deletions python-sdk/src/astro/sql/operators/raw_sql.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,10 @@
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
Expand Down
Loading