Skip to content

Commit 9db12f7

Browse files
Move aio cursors under pyathena/aio/ package
Relocate specialized aio cursors from pyathena/{pandas,arrow,polars}/ to pyathena/aio/{pandas,arrow,polars}/ to keep all asyncio code isolated under the aio package. This provides clearer separation between sync and async implementations. - pyathena/pandas/aio_cursor.py -> pyathena/aio/pandas/cursor.py - pyathena/arrow/aio_cursor.py -> pyathena/aio/arrow/cursor.py - pyathena/polars/aio_cursor.py -> pyathena/aio/polars/cursor.py Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 0c27362 commit 9db12f7

File tree

10 files changed

+12
-6
lines changed

10 files changed

+12
-6
lines changed

pyathena/aio/arrow/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# -*- coding: utf-8 -*-
2+
from __future__ import annotations

pyathena/aio/pandas/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# -*- coding: utf-8 -*-
2+
from __future__ import annotations

pyathena/aio/polars/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# -*- coding: utf-8 -*-
2+
from __future__ import annotations

tests/pyathena/aio/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ async def aio_dict_cursor(request):
4242

4343
@pytest.fixture
4444
async def aio_pandas_cursor(request):
45-
from pyathena.pandas.aio_cursor import AioPandasCursor
45+
from pyathena.aio.pandas.cursor import AioPandasCursor
4646

4747
if not hasattr(request, "param"):
4848
setattr(request, "param", {}) # noqa: B010
@@ -56,7 +56,7 @@ async def aio_pandas_cursor(request):
5656

5757
@pytest.fixture
5858
async def aio_arrow_cursor(request):
59-
from pyathena.arrow.aio_cursor import AioArrowCursor
59+
from pyathena.aio.arrow.cursor import AioArrowCursor
6060

6161
if not hasattr(request, "param"):
6262
setattr(request, "param", {}) # noqa: B010
@@ -70,7 +70,7 @@ async def aio_arrow_cursor(request):
7070

7171
@pytest.fixture
7272
async def aio_polars_cursor(request):
73-
from pyathena.polars.aio_cursor import AioPolarsCursor
73+
from pyathena.aio.polars.cursor import AioPolarsCursor
7474

7575
if not hasattr(request, "param"):
7676
setattr(request, "param", {}) # noqa: B010

tests/pyathena/arrow/test_aio_cursor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ async def test_no_result_set_raises(self, aio_arrow_cursor):
5656
aio_arrow_cursor.as_polars()
5757

5858
async def test_context_manager(self):
59-
from pyathena.arrow.aio_cursor import AioArrowCursor
59+
from pyathena.aio.arrow.cursor import AioArrowCursor
6060

6161
conn = await _aio_connect(schema_name=ENV.schema, cursor_class=AioArrowCursor)
6262
try:

tests/pyathena/pandas/test_aio_cursor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ async def test_no_result_set_raises(self, aio_pandas_cursor):
4848
aio_pandas_cursor.as_pandas()
4949

5050
async def test_context_manager(self):
51-
from pyathena.pandas.aio_cursor import AioPandasCursor
51+
from pyathena.aio.pandas.cursor import AioPandasCursor
5252

5353
conn = await _aio_connect(schema_name=ENV.schema, cursor_class=AioPandasCursor)
5454
try:

tests/pyathena/polars/test_aio_cursor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ async def test_no_result_set_raises(self, aio_polars_cursor):
5555
aio_polars_cursor.as_arrow()
5656

5757
async def test_context_manager(self):
58-
from pyathena.polars.aio_cursor import AioPolarsCursor
58+
from pyathena.aio.polars.cursor import AioPolarsCursor
5959

6060
conn = await _aio_connect(schema_name=ENV.schema, cursor_class=AioPolarsCursor)
6161
try:

0 commit comments

Comments
 (0)