refactor: polywrap-msgpack package (#180)

Author: Niraj-Kamdar

packages/polywrap-msgpack/polywrap_msgpack/__init__.py

@@ -14,13 +14,18 @@
 from .sanitize import *
 
 __all__ = [
+    # Serializer
     "msgpack_decode",
     "msgpack_encode",
+    # Extensions
     "decode_ext_hook",
     "encode_ext_hook",
-    "sanitize",
     "ExtensionTypes",
+    # Sanitizers
+    "sanitize",
+    # Extention types
     "GenericMap",
+    # Errors
     "MsgpackError",
     "MsgpackDecodeError",
     "MsgpackEncodeError",

packages/polywrap-msgpack/polywrap_msgpack/decoder.py

@@ -43,8 +43,8 @@ def msgpack_decode(val: bytes) -> Any:
         Any: any python object
     """
     try:
-        return msgpack.unpackb(
+        return msgpack.unpackb(  # pyright: ignore[reportUnknownMemberType]
             val, ext_hook=decode_ext_hook
-        )  # pyright: reportUnknownMemberType=false
+        )
     except Exception as e:
         raise MsgpackDecodeError("Failed to decode msgpack data") from e

packages/polywrap-msgpack/polywrap_msgpack/encoder.py

@@ -28,8 +28,12 @@ def encode_ext_hook(obj: Any) -> ExtType:
         return ExtType(
             ExtensionTypes.GENERIC_MAP.value,
             # pylint: disable=protected-access
-            msgpack_encode(cast(GenericMap[Any, Any], obj)._map),
-        )  # pyright: reportPrivateUsage=false
+            msgpack_encode(
+                cast(
+                    GenericMap[Any, Any], obj
+                )._map  # pyright: ignore[reportPrivateUsage]
+            ),
+        )
     raise MsgpackExtError(f"Object of type {type(obj)} is not supported")
 
 

packages/polywrap-msgpack/polywrap_msgpack/sanitize.py

@@ -22,7 +22,7 @@ def sanitize(value: Any) -> Any:
     if isinstance(value, GenericMap):
         dictionary: Dict[Any, Any] = cast(
             GenericMap[Any, Any], value
-        )._map  # pyright: reportPrivateUsage=false
+        )._map  # pyright: ignore[reportPrivateUsage]
         new_map: GenericMap[str, Any] = GenericMap({})
         for key, val in dictionary.items():
             if not isinstance(key, str):