Minor typing fixes (#87)

pydicom · Jun 17, 2024 · bb3108b · bb3108b
1 parent d78f585
commit bb3108b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/pylibjpeg/utils.py b/pylibjpeg/utils.py
@@ -32,7 +32,7 @@ def __call__(self, src: bytes, **kwargs: Any) -> np.ndarray:
 
 
 class Encoder(Protocol):
-    def __call__(self, src: np.ndarray, **kwargs: Any) -> Union[bytes, bytearray]:
+    def __call__(self, src: Union[np.ndarray, bytes], **kwargs: Any) -> Union[bytes, bytearray]:
         ...  # pragma: no cover
 
 
@@ -347,7 +347,10 @@ def get_pixel_data_decoders(
 
     entry_point = "pylibjpeg.pixel_data_decoders"
     decoders = cast(
-        Union[Dict[str, Decoder], Dict[str, Dict[str, Decoder]]],
+        Union[
+            Dict[str, Union[Decoder, Encoder]],
+            Dict[str, Dict[str, Union[Decoder, Encoder]]],
+        ],
         _get_pixel_data_plugins(entry_point, version),
     )
     return decoders

diff --git a/pyproject.toml b/pyproject.toml
@@ -32,7 +32,7 @@ license = {text = "MIT"}
 name = "pylibjpeg"
 readme = "README.md"
 requires-python = ">=3.8"
-version = "2.0.0"
+version = "2.0.1"
 
 
 [project.optional-dependencies]