Skip to content

Commit

Permalink
Patch codecov for fully_bayesian_multitask (pytorch#2295)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#2295

--

Reviewed By: Balandat

Differential Revision: D56050007

fbshipit-source-id: 8e6e8cf222faf6f336cac4e6c377221b510d061c
  • Loading branch information
saitcakmak authored and facebook-github-bot committed Apr 12, 2024
1 parent dc219ca commit 59ec39e
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions test/models/test_fully_bayesian_multitask.py
Original file line number Diff line number Diff line change
Expand Up @@ -166,6 +166,15 @@ def test_raises(self):
task_feature=4,
)
train_X, train_Y, train_Yvar, model = self._get_data_and_model(**tkwargs)
with self.assertRaisesRegex(
NotImplementedError, "`all_tasks` argument is not supported"
):
SaasFullyBayesianMultiTaskGP(
train_X=train_X,
train_Y=train_Y,
task_feature=-1,
all_tasks=[0, 1, 2, 3],
)
sampler = IIDNormalSampler(sample_shape=torch.Size([2]))
with self.assertRaisesRegex(
NotImplementedError, "Fantasize is not implemented!"
Expand Down

0 comments on commit 59ec39e

Please sign in to comment.