From 7c8e4b3e6be09ffd1ce534523ee36fd500a09d26 Mon Sep 17 00:00:00 2001
From: Maxim Ziatdinov <34245227+ziatdinovmax@users.noreply.github.com>
Date: Tue, 10 Oct 2023 22:27:36 -0400
Subject: [PATCH] Update docstring

---
 gpax/models/hskgp.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/gpax/models/hskgp.py b/gpax/models/hskgp.py
index 25c7a90..e3d693c 100644
--- 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: