Fix np.ndarray typing

anibali · Mar 13, 2024 · 5d5404c · 5d5404c
1 parent 3fcd2ca
commit 5d5404c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/webp/__init__.py b/webp/__init__.py
@@ -509,7 +509,7 @@ def new(webp_data: WebPData, dec_opts: Optional[WebPAnimDecoderOptions] = None)
 
 def imwrite(
         file_path: str,
-        arr: "np.ndarray[Any, Any]",
+        arr: "np.ndarray[Any, np.dtype[np.uint8]]",
         *args: Any,
         pilmode: Optional[str] = None,
         **kwargs: Any) -> None:
@@ -573,7 +573,7 @@ def _mimwrite_pics(
 
 def mimwrite(
         file_path: str,
-        arrs: List[np.ndarray[Any, Any]],
+        arrs: "List[np.ndarray[Any, np.dtype[np.uint8]]]",
         *args: Any,
         fps: float = 30.0,
         pilmode: Optional[str] = None,