This is an automated email from the ASF dual-hosted git repository.
lidavidm pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/arrow-adbc.git
The following commit(s) were added to refs/heads/main by this push:
new 47e9e6b10 fix(python/adbc_driver_manager): update type annotations
(#3603)
47e9e6b10 is described below
commit 47e9e6b106aa3f3ffe2f8ab3f60874a0725199da
Author: David Li <[email protected]>
AuthorDate: Mon Oct 27 15:58:40 2025 +0900
fix(python/adbc_driver_manager): update type annotations (#3603)
---
python/adbc_driver_manager/adbc_driver_manager/_lib.pyi | 13 +++++++++----
.../tests/test_dbapi_polars_nopyarrow.py | 2 +-
2 files changed, 10 insertions(+), 5 deletions(-)
diff --git a/python/adbc_driver_manager/adbc_driver_manager/_lib.pyi
b/python/adbc_driver_manager/adbc_driver_manager/_lib.pyi
index 33d234e82..fe42c0d92 100644
--- a/python/adbc_driver_manager/adbc_driver_manager/_lib.pyi
+++ b/python/adbc_driver_manager/adbc_driver_manager/_lib.pyi
@@ -17,13 +17,13 @@
# NOTE: generated with mypy's stubgen, then hand-edited to fix things
-import typing_extensions
-from typing import Any, ClassVar, List, Literal, Optional, Tuple, Union
-
-from typing import overload
import enum
import pathlib
import typing
+from typing import overload, Any, ClassVar, List, Literal, Optional, Tuple,
Union
+
+import typing_extensions
+from typing_extensions import CapsuleType
INGEST_OPTION_MODE: str
INGEST_OPTION_MODE_APPEND: str
@@ -150,11 +150,16 @@ class ArrowArrayStreamHandle:
address: int
is_valid: bool
def release(self) -> None: ...
+ def __arrow_c_schema__(self) -> CapsuleType: ...
+ def __arrow_c_stream__(
+ self, requested_schema=Optional[CapsuleType]
+ ) -> CapsuleType: ...
class ArrowSchemaHandle:
address: int
is_valid: bool
def release(self) -> None: ...
+ def __arrow_c_schema__(self) -> CapsuleType: ...
class DataError(DatabaseError): ...
class DatabaseError(Error): ...
diff --git a/python/adbc_driver_manager/tests/test_dbapi_polars_nopyarrow.py
b/python/adbc_driver_manager/tests/test_dbapi_polars_nopyarrow.py
index 7bf7476f2..43f4d292f 100644
--- a/python/adbc_driver_manager/tests/test_dbapi_polars_nopyarrow.py
+++ b/python/adbc_driver_manager/tests/test_dbapi_polars_nopyarrow.py
@@ -83,7 +83,7 @@ def test_query(sqlite: dbapi.Connection) -> None:
with sqlite.cursor() as cursor:
cursor.execute("SELECT 1 AS theresult")
capsule = cursor.fetch_arrow()
- df = typing.cast(polars.DataFrame, polars.from_arrow(capsule))
+ df = polars.from_arrow(capsule)
polars.testing.assert_frame_equal(
df,
polars.DataFrame(
