gpjax 0.13.1__tar.gz → 0.13.2__tar.gz

{gpjax-0.13.1 → gpjax-0.13.2}/.github/workflows/pr_greeting.yml

@@ -14,7 +14,30 @@ jobs:
       - uses: actions/first-interaction@v3.1.0
         with:
           repo_token: ${{ secrets.GITHUB_TOKEN }}
-          pr_message: >+
+          issue_message: |
+            Thank you for opening your first issue into GPJax!
+
+            If you have not heard from us in a while, please feel free to ping
+            `@gpjax/developers` or anyone who has commented on the PR.
+            Most of our reviewers are volunteers and sometimes things fall
+            through the cracks.
+
+
+            You can also join us [on
+            Slack](https://join.slack.com/t/gpjax/shared_invite/zt-1da57pmjn-rdBCVg9kApirEEn2E5Q2Zw) for real-time
+            discussion.
+
+
+            For details on testing, writing docs, and our review process,
+            please see [the developer
+            guide](https://docs.jaxgaussianprocesses.com/contributing/)
+
+
+            We strive to be a welcoming and open project. Please follow our
+            [Code of
+            Conduct](https://github.com/JaxGaussianProcesses/GPJax/blob/main/.github/CODE_OF_CONDUCT.md).
+
+          pr_message: |
             Thank you for opening your first PR into GPJax!
 
 

{gpjax-0.13.1 → gpjax-0.13.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpjax
-Version: 0.13.1
+Version: 0.13.2
 Summary: Gaussian processes in JAX.
 Project-URL: Documentation, https://docs.jaxgaussianprocesses.com/
 Project-URL: Issues, https://github.com/JaxGaussianProcesses/GPJax/issues

{gpjax-0.13.1 → gpjax-0.13.2}/gpjax/__init__.py

@@ -40,7 +40,7 @@ __license__ = "MIT"
 __description__ = "Gaussian processes in JAX and Flax"
 __url__ = "https://github.com/JaxGaussianProcesses/GPJax"
 __contributors__ = "https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors"
-__version__ = "0.13.1"
+__version__ = "0.13.2"
 
 __all__ = [
     "gps",

{gpjax-0.13.1 → gpjax-0.13.2}/gpjax/kernels/computations/basis_functions.py

@@ -6,9 +6,7 @@ from jaxtyping import Float
 import gpjax
 from gpjax.kernels.computations.base import AbstractKernelComputation
 from gpjax.linalg import (
-    Dense,
     Diagonal,
-    psd,
 )
 from gpjax.typing import Array
 
@@ -27,9 +25,9 @@ class BasisFunctionComputation(AbstractKernelComputation):
         z2 = self.compute_features(kernel, y)
         return self.scaling(kernel) * jnp.matmul(z1, z2.T)
 
-    def _gram(self, kernel: K, inputs: Float[Array, "N D"]) -> Dense:
+    def _gram(self, kernel: K, inputs: Float[Array, "N D"]) -> Float[Array, "N N"]:
         z1 = self.compute_features(kernel, inputs)
-        return psd(Dense(self.scaling(kernel) * jnp.matmul(z1, z1.T)))
+        return self.scaling(kernel) * jnp.matmul(z1, z1.T)
 
     def diagonal(self, kernel: K, inputs: Float[Array, "N D"]) -> Diagonal:
         r"""For a given kernel, compute the elementwise diagonal of the