Update gpgpytorch.py

Fixed a type with respect to noise priors
luinardi · Jan 23, 2024 · 5d8ddd3 · 5d8ddd3
1 parent 976b1b9
commit 5d8ddd3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/hypermapper/bo/models/gpgpytorch.py b/hypermapper/bo/models/gpgpytorch.py
@@ -38,13 +38,13 @@ def __init__(
         else:
             # define noise priors
             if settings["noise_prior"]["name"] == "gamma":
-                alpha = float(settings["lengthscale_prior"]["parameters"][0])
-                beta = float(settings["lengthscale_prior"]["parameters"][1])
+                alpha = float(settings["noise_prior"]["parameters"][0])
+                beta = float(settings["noise_prior"]["parameters"][1])
                 noise_prior = GammaPrior(concentration=alpha, rate=beta)
 
             elif settings["noise_prior"]["name"] == "lognormal":
-                mu = float(settings["lengthscale_prior"]["parameters"][0])
-                sigma = float(settings["lengthscale_prior"]["parameters"][1])
+                mu = float(settings["noise_prior"]["parameters"][0])
+                sigma = float(settings["noise_prior"]["parameters"][1])
                 noise_prior = LogNormalPrior(loc=mu, scale=sigma)
 
             else: