Merge branch 'aevri/psutil_affinity' into 'master'

_platform.get_cpu_count: use psutil instead of os

See merge request BuildStream/buildstream!1394

1 files changed, 3 insertions, 1 deletions

diff --git a/src/buildstream/_platform/platform.py b/src/buildstream/_platform/platform.py
index dba60ddca..0fbe145f9 100644
--- a/src/buildstream/_platform/platform.py
+++ b/src/buildstream/_platform/platform.py
@@ -22,6 +22,8 @@ import platform
 import sys
 import resource
 
+import psutil
+
 from .._exceptions import PlatformError, ImplError
 
 
@@ -68,7 +70,7 @@ class Platform():
         return cls._instance
 
     def get_cpu_count(self, cap=None):
-        cpu_count = len(os.sched_getaffinity(0))
+        cpu_count = len(psutil.Process().cpu_affinity())
         if cap is None:
             return cpu_count
         else: