diff --git a/sdk/python/feast/client.py b/sdk/python/feast/client.py
index c89c462090..b6c1b45f9e 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
@@ -93,7 +94,7 @@
 
 _logger = logging.getLogger(__name__)
 
-CPU_COUNT: int = len(os.sched_getaffinity(0))
+CPU_COUNT: int = multiprocessing.cpu_count()
 
 warnings.simplefilter("once", DeprecationWarning)