diff --git a/pyproject.toml b/pyproject.toml index eaa5cdd509..efe8818d17 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,6 +58,7 @@ dependencies = [ "h5py>=3.1", "tqdm", "scikit-learn>=0.24", + "scikit-learn-intelex>=2024.3.0", "statsmodels>=0.13", "patsy", "networkx>=2.7", diff --git a/scanpy/tools/_tsne.py b/scanpy/tools/_tsne.py index 1e3ade92e6..6651f57b84 100644 --- a/scanpy/tools/_tsne.py +++ b/scanpy/tools/_tsne.py @@ -156,6 +156,8 @@ def tsne( ) ) if use_fast_tsne is False: # In case MultiCore failed to import + from sklearnex import patch_sklearn,unpatch_sklearn + patch_sklearn() from sklearn.manifold import TSNE # unfortunately, sklearn does not allow to set a minimum number @@ -163,7 +165,7 @@ def tsne( tsne = TSNE(**params_sklearn) logg.info(" using sklearn.manifold.TSNE") X_tsne = tsne.fit_transform(X) - + unpatch_sklearn() # update AnnData instance adata.obsm["X_tsne"] = X_tsne # annotate samples with tSNE coordinates adata.uns["tsne"] = {