tests

Author: odunbar

test/GaussianProcess/runtests.jl

@@ -96,7 +96,6 @@ using CalibrateEmulateSample.Utilities
     # skip rebuild:
     @test_logs (:warn,) Emulator(agp, iopairs, encoder_schedule = [], kernel_params = kernel_params)
 
-
     μ1b, σ1b² = Emulators.predict(em_agp_from_gp1, new_inputs)
 
     # gp1 and agp_from_gp2 should give similar predictions
@@ -112,7 +111,7 @@ using CalibrateEmulateSample.Utilities
     gp2 = GaussianProcess(gppackage; kernel = GPkernel, noise_learn = true, prediction_type = pred_type)
 
     em2 = Emulator(gp2, iopairs, encoder_schedule = [])
-
+    
     Emulators.optimize_hyperparameters!(em2)
 
     μ2, σ2² = Emulators.predict(em2, new_inputs)
@@ -187,8 +186,18 @@ using CalibrateEmulateSample.Utilities
 
     em4_noise_from_Σ = Emulator(gp4, iopairs2; encoder_kwargs = (; obs_noise_cov = Σ))
 
-    Emulators.optimize_hyperparameters!(em4_noise_learnt)
-    Emulators.optimize_hyperparameters!(em4_noise_from_Σ)
+    # add some kernel bounds
+    n_hparams = length(Emulators.get_params(gp4)[1])
+    low = repeat([log(1e-4)], n_hparams) # bounds provided in log space
+    high = repeat([log(1e4)], n_hparams)
+    kernbounds = (low, high)
+    n_hparams2 = length(Emulators.get_params(gp4_noise_learnt)[1])
+    low2 = repeat([log(1e-4)], n_hparams2) # bounds provided in log space
+    high2 = repeat([log(1e4)], n_hparams2)
+    kernbounds2 = (low2, high2)
+    
+    Emulators.optimize_hyperparameters!(em4_noise_from_Σ; kernbounds=kernbounds)
+    Emulators.optimize_hyperparameters!(em4_noise_learnt; kernbounds=kernbounds2)
 
     new_inputs = zeros(2, 4)
     new_inputs[:, 2] = [π / 2, π]