just unbreak installation with setuptools and guide users towards manual pip install, if user hubert

Author: lucidrains

audiolm_pytorch/hubert_kmeans.py

@@ -36,6 +36,10 @@ def __init__(
         output_layer = 9
     ):
         super().__init__()
+
+        import sklearn
+        assert sklearn.__version__ == '0.24.0', 'scikit-learn needs to be exactly 0.24.0 - please install the correct version by running `pip install scikit-learn==0.24.0`'
+
         self.target_sample_hz = target_sample_hz
         self.seq_len_multiple_of = seq_len_multiple_of
         self.output_layer = output_layer

audiolm_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.2.12'
+__version__ = '1.2.14'

setup.py

@@ -28,7 +28,7 @@
     'joblib',
     'lion-pytorch',
     'local-attention>=1.8.4',
-    'scikit-learn==0.24.0',
+    'scikit-learn',
     'sentencepiece',
     'torch>=1.12',
     'torchaudio',