-
Notifications
You must be signed in to change notification settings - Fork 53
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Cleanup src #419
Cleanup src #419
Conversation
docs/examples/graph_kernels.py
Outdated
@@ -158,6 +174,7 @@ | |||
model=posterior, | |||
objective=jit(gpx.ConjugateMLL(negative=True)), | |||
train_data=D, | |||
max_iters=1000 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
did this fix it? Seems like a huge number for LBFGS
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
No - the error persists. Works locally
covariance = Ktt - jnp.matmul(Kxt.T, Sigma_inv_Kxt) | ||
covariance += cola.ops.I_like(covariance) * self.prior.jitter | ||
covariance = cola.PSD(covariance) | ||
rval = GaussianDistribution(jnp.atleast_1d(mean.squeeze()), covariance) | ||
if m == 1: | ||
return rval | ||
else: | ||
return ReshapedDistribution(rval, (n_test, m)) | ||
|
||
return GaussianDistribution(jnp.atleast_1d(mean.squeeze()), covariance) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Readable!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice bit of surgery Dr. Pinder
Type of changes
Checklist
poetry run pre-commit run --all-files --show-diff-on-failure
before committing.Description
Tighten up the underlying codebase by removing aspects of the code that are either incongruent with the entire API, or introduce too much boilerplate code for their returned utility. The code being removed is deeply woven into the source code, so adding deprecations is not possible.
Notes:
strict=True
argument has been added to allzip
calls to make Ruff compliant with Python 3.10 standards. See B905Issue Number: N/A