diff --git a/hdbscan/tests/test_hdbscan.py b/hdbscan/tests/test_hdbscan.py index 05bf6932..770a8904 100644 --- a/hdbscan/tests/test_hdbscan.py +++ b/hdbscan/tests/test_hdbscan.py @@ -25,7 +25,7 @@ membership_vector, all_points_membership_vectors) # from sklearn.cluster.tests.common import generate_clustered_data -from sklearn.datasets import make_blobs +from sklearn.datasets import make_blobs, make_moons from sklearn.utils import shuffle from sklearn.preprocessing import StandardScaler from scipy.stats import mode @@ -34,8 +34,6 @@ from functools import wraps from nose import SkipTest -from sklearn import datasets - import warnings n_clusters = 3 @@ -94,10 +92,10 @@ def run_test(*args, **kwargs): def generate_noisy_data(): - blobs, _ = datasets.make_blobs(n_samples=200, - centers=[(-0.75, 2.25), (1.0, 2.0)], - cluster_std=0.25) - moons, _ = datasets.make_moons(n_samples=200, noise=0.05) + blobs, _ = make_blobs(n_samples=200, + centers=[(-0.75, 2.25), (1.0, 2.0)], + cluster_std=0.25) + moons, _ = make_moons(n_samples=200, noise=0.05) noise = np.random.uniform(-1.0, 3.0, (50, 2)) return np.vstack([blobs, moons, noise])