diff --git a/test/acquisition/test_fixed_feature.py b/test/acquisition/test_fixed_feature.py index 398f86370b..d4d3f47059 100644 --- a/test/acquisition/test_fixed_feature.py +++ b/test/acquisition/test_fixed_feature.py @@ -83,6 +83,9 @@ def test_fixed_features(self): ) qEI_ff.set_X_pending(X_pending[..., :-1]) self.assertTrue(torch.allclose(qEI.X_pending, X_pending)) + # test setting to None + qEI_ff.X_pending = None + self.assertIsNone(qEI_ff.X_pending) # test gradient test_X = torch.rand(1, 3, device=self.device, requires_grad=True)