23 lines
615 B
Python
23 lines
615 B
Python
|
import sys
|
||
|
|
|
||
|
|
import numba
|
||
|
|
|
||
|
|
from .pynndescent_ import NNDescent, PyNNDescentTransformer
|
||
|
|
|
||
|
|
if sys.version_info[:2] >= (3, 8):
|
||
|
|
import importlib.metadata as importlib_metadata
|
||
|
|
else:
|
||
|
|
import importlib_metadata
|
||
|
|
|
||
|
|
# Workaround: https://github.com/numba/numba/issues/3341
|
||
|
|
if numba.config.THREADING_LAYER == "omp":
|
||
|
|
try:
|
||
|
|
from numba.np.ufunc import tbbpool
|
||
|
|
|
||
|
|
numba.config.THREADING_LAYER = "tbb"
|
||
|
|
except ImportError as e:
|
||
|
|
# might be a missing symbol due to e.g. tbb libraries missing
|
||
|
|
numba.config.THREADING_LAYER = "workqueue"
|
||
|
|
|
||
|
|
__version__ = importlib_metadata.version("pynndescent")
|