From 0ab7112c981397a8b2c992311f81f3b5dadfaedb Mon Sep 17 00:00:00 2001 From: "Michael R. Crusoe" Date: Sun, 7 Dec 2025 13:12:51 +0100 Subject: [PATCH 1/3] [psutil] add overloads for cpu_affinity --- stubs/psutil/psutil/__init__.pyi | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/stubs/psutil/psutil/__init__.pyi b/stubs/psutil/psutil/__init__.pyi index 3e23309aadda..ee71415e2d8c 100644 --- a/stubs/psutil/psutil/__init__.pyi +++ b/stubs/psutil/psutil/__init__.pyi @@ -198,6 +198,11 @@ class Process: if sys.platform != "darwin": def io_counters(self) -> pio: ... def ionice(self, ioclass: int | None = None, value: int | None = None) -> pionice: ... + @overload + def cpu_affinity(self, cpus: None) -> list[int]: ... + @overload + def cpu_affinity(self, cpus: list[int]) -> None: ... + @overload def cpu_affinity(self, cpus: list[int] | None = None) -> list[int] | None: ... def memory_maps(self, grouped: bool = True): ... if sys.platform == "linux": From 8a12b510db86192c48d301aa13b116cbd01d72e6 Mon Sep 17 00:00:00 2001 From: "Michael R. Crusoe" <1330696+mr-c@users.noreply.github.com> Date: Tue, 9 Dec 2025 19:25:17 +0100 Subject: [PATCH 2/3] remove 3rd overload Co-authored-by: Semyon Moroz --- stubs/psutil/psutil/__init__.pyi | 2 -- 1 file changed, 2 deletions(-) diff --git a/stubs/psutil/psutil/__init__.pyi b/stubs/psutil/psutil/__init__.pyi index ee71415e2d8c..417a1458b7ca 100644 --- a/stubs/psutil/psutil/__init__.pyi +++ b/stubs/psutil/psutil/__init__.pyi @@ -202,8 +202,6 @@ class Process: def cpu_affinity(self, cpus: None) -> list[int]: ... @overload def cpu_affinity(self, cpus: list[int]) -> None: ... - @overload - def cpu_affinity(self, cpus: list[int] | None = None) -> 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]: ... From 68582132c3b7b1c038e778f7a6f7f7ebbf919e1b Mon Sep 17 00:00:00 2001 From: "Michael R. Crusoe" Date: Tue, 9 Dec 2025 20:24:37 +0100 Subject: [PATCH 3/3] default value --- stubs/psutil/psutil/__init__.pyi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stubs/psutil/psutil/__init__.pyi b/stubs/psutil/psutil/__init__.pyi index 417a1458b7ca..d3643c6568b5 100644 --- a/stubs/psutil/psutil/__init__.pyi +++ b/stubs/psutil/psutil/__init__.pyi @@ -199,7 +199,7 @@ class Process: def io_counters(self) -> pio: ... def ionice(self, ioclass: int | None = None, value: int | None = None) -> pionice: ... @overload - def cpu_affinity(self, cpus: None) -> list[int]: ... + 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): ...