fix: add missing submodule benchmark, csv and cuda (#71)

zen-xu · Sep 6, 2024 · 1b4b5f2 · 1b4b5f2
1 parent 2aad904
commit 1b4b5f2
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 0 deletions.
diff --git a/pyarrow-stubs/benchmark.pyi b/pyarrow-stubs/benchmark.pyi
@@ -0,0 +1,3 @@
+from pyarrow.lib import benchmark_PandasObjectIsNull
+
+__all__ = ["benchmark_PandasObjectIsNull"]
diff --git a/pyarrow-stubs/csv.pyi b/pyarrow-stubs/csv.pyi
@@ -0,0 +1,27 @@
+from pyarrow._csv import (
+    ISO8601,
+    ConvertOptions,
+    CSVStreamingReader,
+    CSVWriter,
+    InvalidRow,
+    ParseOptions,
+    ReadOptions,
+    WriteOptions,
+    open_csv,
+    read_csv,
+    write_csv,
+)
+
+__all__ = [
+    "ISO8601",
+    "ConvertOptions",
+    "CSVStreamingReader",
+    "CSVWriter",
+    "InvalidRow",
+    "ParseOptions",
+    "ReadOptions",
+    "WriteOptions",
+    "open_csv",
+    "read_csv",
+    "write_csv",
+]
diff --git a/pyarrow-stubs/cuda.pyi b/pyarrow-stubs/cuda.pyi
@@ -0,0 +1,25 @@
+from pyarrow._cuda import (
+    BufferReader,
+    BufferWriter,
+    Context,
+    CudaBuffer,
+    HostBuffer,
+    IpcMemHandle,
+    new_host_buffer,
+    read_message,
+    read_record_batch,
+    serialize_record_batch,
+)
+
+__all__ = [
+    "BufferReader",
+    "BufferWriter",
+    "Context",
+    "CudaBuffer",
+    "HostBuffer",
+    "IpcMemHandle",
+    "new_host_buffer",
+    "read_message",
+    "read_record_batch",
+    "serialize_record_batch",
+]