diff --git a/botorch/acquisition/utils.py b/botorch/acquisition/utils.py index 86fede0e38..f0ce219e69 100644 --- a/botorch/acquisition/utils.py +++ b/botorch/acquisition/utils.py @@ -149,6 +149,7 @@ def get_acquisition_function( sampler=sampler, objective=objective, constraints=constraints, + X_pending=X_pending, ) raise NotImplementedError( f"Unknown acquisition function {acquisition_function_name}" diff --git a/test/acquisition/test_utils.py b/test/acquisition/test_utils.py index f1c505bb91..c7d8b3aeb1 100644 --- a/test/acquisition/test_utils.py +++ b/test/acquisition/test_utils.py @@ -344,6 +344,7 @@ def test_GetQEHVI(self, mock_acqf): ref_point=self.ref_point, partitioning=mock.ANY, sampler=mock.ANY, + X_pending=self.X_pending, ) args, kwargs = mock_acqf.call_args self.assertEqual(args, ())