Dev/699 win build fix (#710)

* resolves #699, win build error * lint with black --------- Co-authored-by: Roland Stevenson <[email protected]>
uber · Nov 17, 2023 · 3615bc8 · 3615bc8
1 parent c154afe
commit 3615bc8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,5 @@
 import multiprocessing as mp
+import os
 from setuptools import dist, setup, find_packages
 from setuptools.extension import Extension
 
@@ -41,8 +42,12 @@
 
 packages = find_packages(exclude=["tests", "tests.*"])
 
+nthreads = mp.cpu_count()
+if os.name == "nt":
+    nthreads = 0
+
 setup(
     packages=packages,
-    ext_modules=cythonize(extensions, annotate=True, nthreads=mp.cpu_count()),
+    ext_modules=cythonize(extensions, annotate=True, nthreads=nthreads),
     include_dirs=[np_get_include()],
 )