Update docstring

ziatdinovmax · Oct 11, 2023 · 7c8e4b3 · 7c8e4b3
1 parent 95abe51
commit 7c8e4b3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gpax/models/hskgp.py b/gpax/models/hskgp.py
@@ -38,6 +38,8 @@ class VarNoiseGP(ExactGP):
             Optional custom priors over main kernel hyperparameters. Use it when passing your custom kernel.
         mean_fn_prior:
             Optional priors over mean function parameters
+        noise_kernel_prior:
+            Optional custom priors over noise kernel hyperparameters. Use it when passing your custom kernel.
         lengthscale_prior_dist:
             Optional custom prior distribution over main kernel lengthscale. Defaults to LogNormal(0, 1).
         noise_mean_fn: