Skip to content

Commit

Permalink
Dev/699 win build fix (uber#710)
Browse files Browse the repository at this point in the history
* resolves uber#699, win build error
* lint with black

---------

Co-authored-by: Roland Stevenson <rolandrmgservices@gmail.com>
  • Loading branch information
ras44 and rolandrmgservices committed Nov 17, 2023
1 parent c154afe commit 3615bc8
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import multiprocessing as mp
import os
from setuptools import dist, setup, find_packages
from setuptools.extension import Extension

Expand Down Expand Up @@ -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()],
)

0 comments on commit 3615bc8

Please sign in to comment.