diff --git a/pymc/gp/hsgp_approx.py b/pymc/gp/hsgp_approx.py index 38726e438b..14490b6f41 100644 --- a/pymc/gp/hsgp_approx.py +++ b/pymc/gp/hsgp_approx.py @@ -683,12 +683,12 @@ def prior_linearized(self, X: TensorLike): psd = self.scale * self.cov_func.power_spectral_density_approx(J) return (phi_cos, phi_sin), psd - def prior( + def prior( # type: ignore self, name: str, X: TensorLike, dims: str | None = None, - ): # type: ignore + ): R""" Returns the (approximate) GP prior distribution evaluated over the input locations `X`. For usage examples, refer to `pm.gp.Latent`.