Add missing return type for Schema.serialize (#123)

zen-xu · Oct 22, 2024 · fcfe788 · fcfe788
1 parent 428d96b
commit fcfe788
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyarrow-stubs/__lib_pxi/types.pyi b/pyarrow-stubs/__lib_pxi/types.pyi
@@ -24,6 +24,7 @@ from pyarrow.lib import (
 )
 from typing_extensions import TypeVar
 
+from .io import Buffer
 from .scalar import ExtensionScalar
 
 _AsPyType = TypeVar("_AsPyType")
@@ -304,7 +305,7 @@ class Schema(_Weakrefable):
     def set(self, i: int, field: Field) -> Schema: ...
     def add_metadata(self, metadata: dict) -> Schema: ...
     def with_metadata(self, metadata: dict) -> Schema: ...
-    def serialize(self, memory_pool: MemoryPool | None = None): ...
+    def serialize(self, memory_pool: MemoryPool | None = None) -> Buffer: ...
     def remove_metadata(self) -> Schema: ...
     def to_string(
         self,