Updating kernel nbs

Author: thomaspinder

docs/nbs/kernels.ipynb

@@ -274,13 +274,15 @@
     "\n",
     "    def __post_init__(self):\n",
     "        self.c = self.period / 2.0  # in [0, \\pi]\n",
-    "        self._params = {\"tau\": jnp.array([4.0])}\n",
     "\n",
     "    def __call__(self, x: f64[\"1 D\"], y: f64[\"1 D\"], params: dict) -> f64[\"1\"]:\n",
     "        tau = params[\"tau\"]\n",
     "        t = angular_distance(x, y, self.c)\n",
     "        K = (1 + tau * t / self.c) * jnp.clip(1 - t / self.c, 0, jnp.inf) ** tau\n",
-    "        return K.squeeze()"
+    "        return K.squeeze()\n",
+    "    \n",
+    "    def _initialise_params(self, key: jr.PRNGKey) -> dict:\n",
+    "        return {\"tau\": jnp.array([4.0])}"
    ]
   },
   {