ENH: Improve io/pickle

Make pickle consistent with upstream pandas
Only include what is in the API

Author: bashtage

pandas-stubs/_typing.pyi

@@ -67,6 +67,9 @@ class BaseBuffer(Protocol): ...
 class ReadBuffer(BaseBuffer, Protocol[AnyStr_cov]): ...
 class WriteBuffer(BaseBuffer, Protocol[AnyStr_cov]): ...
 
+class ReadPickleBuffer(ReadBuffer[bytes], Protocol):
+    def readline(self) -> AnyStr_cov: ...
+
 FilePath = Union[str, PathLike[str]]
 
 Buffer = Union[IO[AnyStr], RawIOBase, BufferedIOBase, TextIOBase, TextIOWrapper, mmap]

pandas-stubs/io/pickle.pyi

@@ -1,16 +1,24 @@
 from __future__ import annotations
 
-from typing import Literal
+from typing import Any
 
-from pandas._typing import FilePathOrBuffer
+from pandas._typing import (
+    CompressionOptions,
+    FilePath,
+    ReadPickleBuffer,
+    StorageOptions,
+    WriteBuffer,
+)
 
 def to_pickle(
-    obj,
-    filepath_or_buffer: FilePathOrBuffer,
-    compression: str | None = ...,
+    obj: Any,
+    filepath_or_buffer: FilePath | WriteBuffer[bytes],
+    compression: CompressionOptions = ...,
     protocol: int = ...,
-): ...
+    storage_options: StorageOptions = ...,
+) -> None: ...
 def read_pickle(
-    filepath_or_buffer_or_reader: FilePathOrBuffer,
-    compression: str | Literal["infer", "gzip", "bz2", "zip", "xz"] | None = ...,
-): ...
+    filepath_or_buffer: FilePath | ReadPickleBuffer,
+    compression: CompressionOptions = ...,
+    storage_options: StorageOptions = ...,
+) -> Any: ...