Commit 75fe56b
Fix PairwiseGP test that fails on CUDA (#3283)
Summary:
Pull Request resolved: #3283
**Context**: `test_load_state_dict_recomputes_laplace` from `botorch.test.models.test_pairwise_gp.TestPairwiseGP` has been failing since it was added (D99173855) due to an issue with tensors on different devices.
**This PR**: Moves a tensor to the appropriate device
Reviewed By: ItsMrLin
Differential Revision: D101675014
fbshipit-source-id: ca91dd5611d706f9b8ddc8ff3269567e9aa62f721 parent 30f2838 commit 75fe56b
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
887 | 887 | | |
888 | 888 | | |
889 | 889 | | |
890 | | - | |
| 890 | + | |
891 | 891 | | |
892 | 892 | | |
893 | 893 | | |
| |||
0 commit comments