From cd347aac55a5cac1a12381aaf0d43654a4e7fba3 Mon Sep 17 00:00:00 2001 From: Trim21 Date: Sat, 21 Sep 2024 12:37:45 +0800 Subject: [PATCH 1/2] fix type --- msgspec/msgpack.pyi | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/msgspec/msgpack.pyi b/msgspec/msgpack.pyi index d1284eae..a0afb3b6 100644 --- a/msgspec/msgpack.pyi +++ b/msgspec/msgpack.pyi @@ -54,7 +54,7 @@ class Decoder(Generic[T]): dec_hook: dec_hook_sig = None, ext_hook: ext_hook_sig = None, ) -> None: ... - def decode(self, data: bytes) -> T: ... + def decode(self, data: bytes | memoryview) -> T: ... class Encoder: enc_hook: enc_hook_sig @@ -76,7 +76,7 @@ class Encoder: @overload def decode( - buf: bytes, + buf: bytes | memoryview, *, strict: bool = True, dec_hook: dec_hook_sig = None, @@ -84,7 +84,7 @@ def decode( ) -> Any: ... @overload def decode( - buf: bytes, + buf: bytes | memoryview, *, type: Type[T] = ..., strict: bool = True, @@ -93,7 +93,7 @@ def decode( ) -> T: ... @overload def decode( - buf: bytes, + buf: bytes | memoryview, *, type: Any = ..., strict: bool = True, From fb3f976c0901e893061d221695f643a60e17ed26 Mon Sep 17 00:00:00 2001 From: Trim21 Date: Sat, 21 Sep 2024 12:43:44 +0800 Subject: [PATCH 2/2] use typing.Buffer --- msgspec/msgpack.pyi | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/msgspec/msgpack.pyi b/msgspec/msgpack.pyi index a0afb3b6..7e77f747 100644 --- a/msgspec/msgpack.pyi +++ b/msgspec/msgpack.pyi @@ -10,6 +10,9 @@ from typing import ( overload, ) +from typing_extensions import Buffer + + T = TypeVar("T") enc_hook_sig = Optional[Callable[[Any], Any]] @@ -54,7 +57,7 @@ class Decoder(Generic[T]): dec_hook: dec_hook_sig = None, ext_hook: ext_hook_sig = None, ) -> None: ... - def decode(self, data: bytes | memoryview) -> T: ... + def decode(self, data: Buffer) -> T: ... class Encoder: enc_hook: enc_hook_sig @@ -76,7 +79,7 @@ class Encoder: @overload def decode( - buf: bytes | memoryview, + buf: Buffer, *, strict: bool = True, dec_hook: dec_hook_sig = None, @@ -84,7 +87,7 @@ def decode( ) -> Any: ... @overload def decode( - buf: bytes | memoryview, + buf: Buffer, *, type: Type[T] = ..., strict: bool = True, @@ -93,7 +96,7 @@ def decode( ) -> T: ... @overload def decode( - buf: bytes | memoryview, + buf: Buffer, *, type: Any = ..., strict: bool = True,