Move sklearn import inside kmeans training function

Author: samanklesaria

examples/hubert/utils/kmeans.py

@@ -8,7 +8,6 @@
 from typing import Tuple
 
 import torch
-from sklearn.cluster import MiniBatchKMeans
 from torch import Tensor
 
 from .common_utils import _get_feat_lens_paths, _get_model_path
@@ -102,6 +101,7 @@ def learn_kmeans(
     """
     if not km_dir.exists():
         km_dir.mkdir()
+    from sklearn.cluster import MiniBatchKMeans
 
     km_model = MiniBatchKMeans(
         n_clusters=n_clusters,

test/torchaudio_unittest/example/hubert/__init__.py

@@ -2,4 +2,4 @@
 import sys
 
 
-# sys.path.append(os.path.join(os.path.dirname(__file__), "..", "..", "..", "..", "examples", "hubert"))
+sys.path.append(os.path.join(os.path.dirname(__file__), "..", "..", "..", "..", "examples", "hubert"))