Skip to content
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

Add batch_shape property to SingleTaskVariationalGP #1799

Closed
wants to merge 1 commit into from

Conversation

Balandat
Copy link
Contributor

This enables the use of SingleTaskVariationalGP with certain botorch features (e.g. with entropy-based acquistion functions as requested in #1795).

This is a bit of a band-aid, the proper thing to do here is to fix up the PR upstreaming this to gpytorch (cornellius-gp/gpytorch#2307) to enable support for batch_shape on all approximate gpytorch models, and then just call that on the model in ApproximateGPyTorchModel.

This enables the use of `SingleTaskVariationalGP` with certain botorch features (e.g. with entropy-based acquistion functions as requested in pytorch#1795).

This is a bit of a band-aid, the proper thing to do here is to fix up the PR upstreaming this to gpytorch (cornellius-gp/gpytorch#2307) to enable support for `batch_shape` on all approximate gpytorch models, and then just call that on the `model` in `ApproximateGPyTorchModel`.
@facebook-github-bot facebook-github-bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Apr 19, 2023
@Balandat
Copy link
Contributor Author

cc @acb-code

@facebook-github-bot
Copy link
Contributor

@Balandat has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@codecov
Copy link

codecov bot commented Apr 19, 2023

Codecov Report

Merging #1799 (0db487e) into main (71690a8) will not change coverage.
The diff coverage is 100.00%.

❗ Current head 0db487e differs from pull request most recent head 8c1efd9. Consider uploading reports for the commit 8c1efd9 to get more accurate results

@@            Coverage Diff            @@
##              main     #1799   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files          170       170           
  Lines        14779     14782    +3     
=========================================
+ Hits         14779     14782    +3     
Impacted Files Coverage Δ
botorch/models/approximate_gp.py 100.00% <100.00%> (ø)

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@facebook-github-bot
Copy link
Contributor

@Balandat merged this pull request in 8bf02a3.

@Balandat Balandat deleted the stvgp_batch_shape branch April 21, 2023 05:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed Do not delete this pull request or issue due to inactivity. Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants