Update docstrings

ziatdinovmax · Feb 7, 2024 · 2ad3fa9 · 2ad3fa9
1 parent a51e695
commit 2ad3fa9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/gpax/models/sparse_gp.py b/gpax/models/sparse_gp.py
@@ -37,8 +37,12 @@ class viSparseGP(viGP):
             Optional custom priors over kernel hyperparameters; uses LogNormal(0,1) by default
         mean_fn_prior:
             Optional priors over mean function parameters
-        noise_prior:
-            Optional custom prior for the observation noise variance; uses LogNormal(0,1) by default.
+        noise_prior_dist:
+            Optional custom prior distribution over the observational noise variance.
+            Defaults to LogNormal(0,1).
+        lengthscale_prior_dist:
+            Optional custom prior distribution over kernel lengthscale.
+            Defaults to LogNormal(0, 1).
         guide:
             Auto-guide option, use 'delta' (default) or 'normal'
     """

diff --git a/gpax/models/vigp.py b/gpax/models/vigp.py
@@ -35,8 +35,12 @@ class viGP(ExactGP):
             Optional custom priors over kernel hyperparameters; uses LogNormal(0,1) by default
         mean_fn_prior:
             Optional priors over mean function parameters
-        noise_prior:
-            Optional custom prior for the observation noise variance; uses LogNormal(0,1) by default.
+        noise_prior_dist:
+            Optional custom prior distribution over the observational noise variance.
+            Defaults to LogNormal(0,1).
+        lengthscale_prior_dist:
+            Optional custom prior distribution over kernel lengthscale.
+            Defaults to LogNormal(0, 1).
         guide:
             Auto-guide option, use 'delta' (default) or 'normal'