From a9d3dca0e88cc672b1f53e3d1499fdfe79406fdb Mon Sep 17 00:00:00 2001
From: pyalex <moskalenko.alexey@gmail.com>
Date: Mon, 29 Jun 2020 15:21:04 +0300
Subject: [PATCH] fix python sdk for non-unix

---
 sdk/python/feast/client.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sdk/python/feast/client.py b/sdk/python/feast/client.py
index 24bc09d79b..8112e1f82d 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: