Try fixing missing pillow types

stubs/D3DShot/METADATA.toml

@@ -1,5 +1,5 @@
 version = "0.1.*"
-requires = ["numpy", "torch", "comtypes"]
+requires = ["numpy", "torch", "comtypes", "types-Pillow"]
 
 [tool.stubtest]
 ignore_missing_stub = false

stubs/D3DShot/d3dshot/capture_output.pyi

@@ -2,10 +2,10 @@ import enum
 from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
-from PIL.Image import Image
+from PIL import Image
 
 _Frame: TypeAlias = Incomplete
-# _Frame: TypeAlias = Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | _Tensor
+# _Frame: TypeAlias = Image.Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | _Tensor
 _Pointer: TypeAlias = Incomplete
 
 class CaptureOutputs(enum.Enum):
@@ -25,5 +25,5 @@ class CaptureOutput:
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> _Frame: ...
-    def to_pil(self, frame: _Frame) -> Image: ...
+    def to_pil(self, frame: _Frame) -> Image.Image: ...
     def stack(self, frames: list[_Frame], stack_dimension) -> _Frame: ...

stubs/D3DShot/d3dshot/capture_outputs/numpy_capture_output.pyi

@@ -2,7 +2,7 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 # TODO: Complete types once we can import non-types dependencies
 # See: https://github.com/python/typeshed/issues/5768
@@ -17,5 +17,5 @@ class NumpyCaptureOutput(CaptureOutput):
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> NDArray: ...
-    def to_pil(self, frame: NDArray) -> Image: ...
+    def to_pil(self, frame: NDArray) -> Image.Image: ...
     def stack(self, frames: list[NDArray] | NDArray, stack_dimension: int) -> NDArray: ...

stubs/D3DShot/d3dshot/capture_outputs/numpy_float_capture_output.pyi

@@ -2,7 +2,7 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 # TODO: Complete types once we can import non-types dependencies
 # See: https://github.com/python/typeshed/issues/5768
@@ -17,5 +17,5 @@ class NumpyFloatCaptureOutput(CaptureOutput):
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> NDArray: ...
-    def to_pil(self, frame: NDArray) -> Image: ...
+    def to_pil(self, frame: NDArray) -> Image.Image: ...
     def stack(self, frames: list[NDArray] | NDArray, stack_dimension: int) -> NDArray: ...

stubs/D3DShot/d3dshot/capture_outputs/pil_capture_output.pyi

@@ -2,14 +2,14 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 _Pointer: TypeAlias = Incomplete
 
 class PILCaptureOutput(CaptureOutput):
     def __init__(self) -> None: ...
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
-    ) -> Image: ...
-    def to_pil(self, frame: Image) -> Image: ...
-    def stack(self, frames: list[Image], stack_dimension: int) -> list[Image]: ...
+    ) -> Image.Image: ...
+    def to_pil(self, frame: Image.Image) -> Image.Image: ...
+    def stack(self, frames: list[Image.Image], stack_dimension: int) -> list[Image.Image]: ...

stubs/D3DShot/d3dshot/capture_outputs/pytorch_capture_output.pyi

@@ -2,7 +2,7 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 # TODO: Complete types once we can import non-types dependencies
 # See: https://github.com/python/typeshed/issues/5768
@@ -15,5 +15,5 @@ class PytorchCaptureOutput(CaptureOutput):
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> _Tensor: ...
-    def to_pil(self, frame: _Tensor) -> Image: ...
+    def to_pil(self, frame: _Tensor) -> Image.Image: ...
     def stack(self, frames: list[_Tensor], stack_dimension: int) -> _Tensor: ...

stubs/D3DShot/d3dshot/capture_outputs/pytorch_float_capture_output.pyi

@@ -2,7 +2,7 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 # TODO: Complete types once we can import non-types dependencies
 # See: https://github.com/python/typeshed/issues/5768
@@ -15,5 +15,5 @@ class PytorchFloatCaptureOutput(CaptureOutput):
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> _Tensor: ...
-    def to_pil(self, frame: _Tensor) -> Image: ...
+    def to_pil(self, frame: _Tensor) -> Image.Image: ...
     def stack(self, frames: list[_Tensor], stack_dimension: int) -> _Tensor: ...

stubs/D3DShot/d3dshot/capture_outputs/pytorch_float_gpu_capture_output.pyi

@@ -2,20 +2,20 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 # TODO: Complete types once we can import non-types dependencies
 # See: https://github.com/python/typeshed/issues/5768
 # from torch import device, Tensor
 _Tensor: TypeAlias = Incomplete
-_device: TypeAlias = Incomplete
+_Device: TypeAlias = Incomplete
 _Pointer: TypeAlias = Incomplete
 
 class PytorchFloatGPUCaptureOutput(CaptureOutput):
-    device: _device
+    device: _Device
     def __init__(self) -> None: ...
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> _Tensor: ...
-    def to_pil(self, frame: _Tensor) -> Image: ...
+    def to_pil(self, frame: _Tensor) -> Image.Image: ...
     def stack(self, frames: list[_Tensor], stack_dimension: int) -> _Tensor: ...

stubs/D3DShot/d3dshot/capture_outputs/pytorch_gpu_capture_output.pyi

@@ -2,21 +2,21 @@ from _typeshed import Incomplete
 from typing_extensions import TypeAlias
 
 from d3dshot.capture_output import CaptureOutput as CaptureOutput
-from PIL.Image import Image
+from PIL import Image
 
 # TODO: Complete types once we can import non-types dependencies
 # See: https://github.com/python/typeshed/issues/5768
 # from torch import device, Tensor
 _Tensor: TypeAlias = Incomplete
-_device: TypeAlias = Incomplete
+_Device: TypeAlias = Incomplete
 
 _Pointer: TypeAlias = Incomplete
 
 class PytorchGPUCaptureOutput(CaptureOutput):
-    device: _device
+    device: _Device
     def __init__(self) -> None: ...
     def process(
         self, pointer: _Pointer, size: int, width: int, height: int, region: tuple[int, int, int, int], rotation: int
     ) -> _Tensor: ...
-    def to_pil(self, frame: _Tensor) -> Image: ...
+    def to_pil(self, frame: _Tensor) -> Image.Image: ...
     def stack(self, frames: list[_Tensor], stack_dimension: int): ...

stubs/D3DShot/d3dshot/d3dshot.pyi

@@ -6,7 +6,7 @@ from d3dshot.capture_output import CaptureOutput as CaptureOutput, CaptureOutput
 from d3dshot.display import Display as Display
 
 _Frame: TypeAlias = Incomplete
-# _Frame: TypeAlias = Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | _Tensor
+# _Frame: TypeAlias = Image.Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | _Tensor
 
 class D3DShot:
     displays: list[Display]

stubs/D3DShot/d3dshot/display.pyi

@@ -1,10 +1,9 @@
 from _typeshed import Incomplete
 from collections.abc import Callable
-from typing import Literal
-from typing_extensions import TypeAlias
+from typing_extensions import Literal, TypeAlias
 
 _Frame: TypeAlias = Incomplete
-# _Frame: TypeAlias = Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | _Tensor
+# _Frame: TypeAlias = Image.Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | _Tensor
 _Pointer: TypeAlias = Incomplete
 
 class Display:

stubs/D3DShot/d3dshot/dll/dxgi.pyi

@@ -10,7 +10,7 @@ from typing_extensions import TypeAlias
 _IUnknown: TypeAlias = Incomplete
 
 _Frame: TypeAlias = Incomplete
-# _Frame: TypeAlias = Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | Tensor
+# _Frame: TypeAlias = Image.Image | npt.NDArray[np.int32] | npt.NDArray[np.float32] | Tensor
 _Pointer: TypeAlias = Incomplete
 
 class LUID(ctypes.Structure): ...