Skip to content

Commit ac7e6fe

Browse files
authored
Merge pull request #594 from gclendenning/set_random_seed
Set seed for generating noisy data
2 parents e55f957 + 6b4675b commit ac7e6fe

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

hdbscan/tests/test_hdbscan.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -118,10 +118,12 @@ def run_test(*args, **kwargs):
118118

119119
def generate_noisy_data():
120120
blobs, _ = datasets.make_blobs(
121-
n_samples=200, centers=[(-0.75, 2.25), (1.0, 2.0)], cluster_std=0.25
121+
n_samples=200, centers=[(-0.75, 2.25), (1.0, 2.0)], cluster_std=0.25, random_state=42
122122
)
123-
moons, _ = datasets.make_moons(n_samples=200, noise=0.05)
124-
noise = np.random.uniform(-1.0, 3.0, (50, 2))
123+
moons, _ = datasets.make_moons(n_samples=200, noise=0.05, random_state=42)
124+
rng = np.random.default_rng(seed=42)
125+
noise = rng.uniform(-1.0, 3.0, (50, 2))
126+
125127
return np.vstack([blobs, moons, noise])
126128

127129

0 commit comments

Comments
 (0)