Skip to content

Commit

Permalink
Remove tbb requirement for non-intel machines
Browse files Browse the repository at this point in the history
tbb is not available on PyPI for non-intel machines. Only include
it as a requirement in the setup.py file for intel machines.

Signed-off-by: Patrick Avery <[email protected]>
  • Loading branch information
psavery committed Jan 16, 2024
1 parent deb9cd5 commit f37cde9
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
Expand Up @@ -2,6 +2,7 @@
import distutils.ccompiler
import os
from pathlib import Path
import platform
from setuptools import setup, find_packages, Extension
import subprocess
import sys
Expand All @@ -24,11 +25,15 @@
'scikit-image',
'scikit-learn',
'scipy',
'tbb',
'tqdm',
'xxhash',
]

if platform.machine() == 'x86_64':
# tbb is an optional dependency, but it is nice to have
# Add it to the requirements automatically for intel computers.
install_reqs.append('tbb')

# This will determine which compiler is being used to build the C modules
compiler_type = distutils.ccompiler.get_default_compiler()
if compiler_type in ("unix", "mingw32"):
Expand Down

0 comments on commit f37cde9

Please sign in to comment.