No module named 'sklearn.cluster._k_means'

[johndpope]

pip install scikit-learn
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
Requirement already satisfied: scikit-learn in /home/jp/miniconda3/envs/torch/lib/python3.8/site-packages (0.24.0)
Requirement already satisfied: scipy>=0.19.1 in /home/jp/miniconda3/envs/torch/lib/python3.8/site-packages (from scikit-learn) (1.5.4)
Requirement already satisfied: threadpoolctl>=2.0.0 in /home/jp/miniconda3/envs/torch/lib/python3.8/site-packages (from scikit-learn) (2.1.0)
Requirement already satisfied: joblib>=0.11 in /home/jp/miniconda3/envs/torch/lib/python3.8/site-packages (from scikit-learn) (1.0.0)
Requirement already satisfied: numpy>=1.13.3 in /home/jp/.local/lib/python3.8/site-packages (from scikit-learn) (1.19.4)
(torch) ➜ RetrieveInStyle git:(main) ✗ pip install scipy
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
Requirement already satisfied: scipy in /home/jp/miniconda3/envs/torch/lib/python3.8/site-packages (1.5.4)
Requirement already satisfied: numpy>=1.14.5 in /home/jp/.local/lib/python3.8/site-packages (from scipy) (1.19.4)

seems like it breaks with latest
https://stackoverflow.com/questions/59881810/modulenotfounderror-no-module-named-sklearn-cluster-k-means-scikit-learn-0-2
needs to be hard coded to older version -

[johndpope]

/home/jp/miniconda3/envs/torch/lib/python3.8/site-packages/sklearn/utils/deprecation.py:144: FutureWarning: The sklearn.cluster.k_means_ module is deprecated in version 0.22 and will be removed in version 0.24. The corresponding classes / functions should instead be imported from sklearn.cluster. Anything that cannot be imported from sklearn.cluster is now part of the private API.
warnings.warn(message, FutureWarning)

sklearn.cluster.k_means_ module ->

sklearn.cluster

[e4s2022]

Just downgrading the scikit-learn to version 0.22 works well for me