From 34776f7934cbdd88329847ae0abd0196e6635915 Mon Sep 17 00:00:00 2001 From: Elizabeth Santorella Date: Thu, 3 Oct 2024 09:21:40 -0700 Subject: [PATCH] Speed up slow input constructor test (#2564) Summary: Pull Request resolved: https://github.com/pytorch/botorch/pull/2564 Differential Revision: D63838676 --- test/acquisition/test_input_constructors.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/acquisition/test_input_constructors.py b/test/acquisition/test_input_constructors.py index 7478ee8a0b..3ad35265b7 100644 --- a/test/acquisition/test_input_constructors.py +++ b/test/acquisition/test_input_constructors.py @@ -109,6 +109,7 @@ from botorch.models.deterministic import FixedSingleSampleModel from botorch.models.model_list_gp_regression import ModelListGP from botorch.sampling.normal import IIDNormalSampler, SobolQMCNormalSampler +from botorch.test_utils.mock import fast_optimize from botorch.utils.constraints import get_outcome_constraint_transforms from botorch.utils.datasets import SupervisedDataset from botorch.utils.multi_objective.box_decompositions.non_dominated import ( @@ -1845,6 +1846,7 @@ def setUp(self, suppress_input_warnings: bool = True) -> None: }, ) + @fast_optimize def test_constructors_can_instantiate(self) -> None: for key, (classes, input_constructor_kwargs) in self.cases.items(): with self.subTest(