1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
|
from warnings import warn, catch_warnings, simplefilter
from .umap_ import UMAP
try:
with catch_warnings():
simplefilter("ignore")
from .parametric_umap import ParametricUMAP
except ImportError:
warn(
"Tensorflow not installed; ParametricUMAP will be unavailable",
category=ImportWarning,
)
# Add a dummy class to raise an error
class ParametricUMAP(object):
def __init__(self, **kwds):
warn(
"""The umap.parametric_umap package requires Tensorflow > 2.0 to be installed.
You can install Tensorflow at https://www.tensorflow.org/install
or you can install the CPU version of Tensorflow using
pip install umap-learn[parametric_umap]
"""
)
raise ImportError(
"umap.parametric_umap requires Tensorflow >= 2.0"
) from None
from .aligned_umap import AlignedUMAP
# Workaround: https://github.com/numba/numba/issues/3341
import numba
import pkg_resources
try:
__version__ = pkg_resources.get_distribution("umap-learn").version
except pkg_resources.DistributionNotFound:
__version__ = "0.5-dev"
|