diff --git a/sdk/python/feast/client.py b/sdk/python/feast/client.py
index 24bc09d79b2..8112e1f82dd 100644
--- a/sdk/python/feast/client.py
+++ b/sdk/python/feast/client.py
@@ -16,6 +16,7 @@
 import datetime
 import logging
 import os
+import multiprocessing
 import shutil
 import tempfile
 import time
@@ -90,7 +91,7 @@
 
 _logger = logging.getLogger(__name__)
 
-CPU_COUNT: int = len(os.sched_getaffinity(0))
+CPU_COUNT: int = multiprocessing.cpu_count()
 
 
 class Client: