From 8be65724868a3fe3f518e4cd6d8219d1241fa3d6 Mon Sep 17 00:00:00 2001
From: "Michael R. Crusoe" <1330696+mr-c@users.noreply.github.com>
Date: Tue, 9 Dec 2025 23:05:04 +0100
Subject: [PATCH] [psutil] add overloads for cpu_affinity (#15112)

---
 stubs/psutil/psutil/__init__.pyi | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/stubs/psutil/psutil/__init__.pyi b/stubs/psutil/psutil/__init__.pyi
index 3e23309aa..d3643c656 100644
--- a/stubs/psutil/psutil/__init__.pyi
+++ b/stubs/psutil/psutil/__init__.pyi
@@ -198,7 +198,10 @@ class Process:
     if sys.platform != "darwin":
         def io_counters(self) -> pio: ...
         def ionice(self, ioclass: int | None = None, value: int | None = None) -> pionice: ...
-        def cpu_affinity(self, cpus: list[int] | None = None) -> list[int] | None: ...
+        @overload
+        def cpu_affinity(self, cpus: None = None) -> list[int]: ...
+        @overload
+        def cpu_affinity(self, cpus: list[int]) -> None: ...
         def memory_maps(self, grouped: bool = True): ...
     if sys.platform == "linux":
         def rlimit(self, resource: int, limits: tuple[int, int] | None = ...) -> tuple[int, int]: ...