diff --git a/ax/storage/botorch_modular_registry.py b/ax/storage/botorch_modular_registry.py index 1d9e6322dd9..7dcc00c2cbc 100644 --- a/ax/storage/botorch_modular_registry.py +++ b/ax/storage/botorch_modular_registry.py @@ -47,6 +47,7 @@ qExpectedHypervolumeImprovement, qNoisyExpectedHypervolumeImprovement, ) +from botorch.acquisition.multi_objective.parego import qLogNParEGO from botorch.acquisition.preference import AnalyticExpectedUtilityOfBestOption from botorch.models import SaasFullyBayesianSingleTaskGP from botorch.models.contextual import LCEAGP @@ -144,6 +145,7 @@ qLogNoisyExpectedImprovement: "qLogNoisyExpectedImprovement", qLogExpectedHypervolumeImprovement: "qLogExpectedHypervolumeImprovement", qLogNoisyExpectedHypervolumeImprovement: "qLogNoisyExpectedHypervolumeImprovement", + qLogNParEGO: "qLogNParEGO", }