Skip to content

Commit 59ec39e

Browse files
saitcakmakfacebook-github-bot
authored andcommitted
Patch codecov for fully_bayesian_multitask (#2295)
Summary: Pull Request resolved: #2295 -- Reviewed By: Balandat Differential Revision: D56050007 fbshipit-source-id: 8e6e8cf222faf6f336cac4e6c377221b510d061c
1 parent dc219ca commit 59ec39e

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

test/models/test_fully_bayesian_multitask.py

+9
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,15 @@ def test_raises(self):
166166
task_feature=4,
167167
)
168168
train_X, train_Y, train_Yvar, model = self._get_data_and_model(**tkwargs)
169+
with self.assertRaisesRegex(
170+
NotImplementedError, "`all_tasks` argument is not supported"
171+
):
172+
SaasFullyBayesianMultiTaskGP(
173+
train_X=train_X,
174+
train_Y=train_Y,
175+
task_feature=-1,
176+
all_tasks=[0, 1, 2, 3],
177+
)
169178
sampler = IIDNormalSampler(sample_shape=torch.Size([2]))
170179
with self.assertRaisesRegex(
171180
NotImplementedError, "Fantasize is not implemented!"

0 commit comments

Comments
 (0)