Skip to content

Commit 4709fe0

Browse files
committed
update tests
1 parent d1c8eda commit 4709fe0

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/emulators/test_gaussian_process_exact.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -167,9 +167,12 @@ def test_gp_corr_deterministic_with_seed(sample_data_y1d, new_data_y1d, device):
167167
x2, _ = new_data_y1d
168168
seed = 42
169169
new_seed = 43
170-
model1 = GaussianProcessCorrelated(x, y, device=device, seed=seed)
171-
model2 = GaussianProcessCorrelated(x, y, device=device, seed=new_seed)
172-
model3 = GaussianProcessCorrelated(x, y, device=device, seed=seed)
170+
set_random_seed(seed)
171+
model1 = GaussianProcessCorrelated(x, y, device=device)
172+
set_random_seed(seed)
173+
model2 = GaussianProcessCorrelated(x, y, device=device)
174+
set_random_seed(new_seed)
175+
model3 = GaussianProcessCorrelated(x, y, device=device)
173176
model1.fit(x, y)
174177
pred1 = model1.predict(x2)
175178
model2.fit(x, y)

0 commit comments

Comments
 (0)