BUG: fix GP prior plot title

elcorto · Dec 14, 2022 · a18bae0 · a18bae0
1 parent d619007
commit a18bae0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/book/content/gp_pred_comp/notebook_plot.py b/book/content/gp_pred_comp/notebook_plot.py
@@ -132,7 +132,7 @@ def calc_gp(
         if noise_level == 0:
             cov_title = r"$\Sigma=K''$"
         else:
-            cov_title = r"$\Sigma=K'' + \sigma_n^2$"
+            cov_title = r"$\Sigma=K'' + \sigma_n^2\,I$"
     else:
         if noise_level == 0:
             cov_title = r"$\Sigma=K'' - K'\,K^{-1}\,K'^\top$"