From 421dd10341d648662478704c7d35b79f2dcd8194 Mon Sep 17 00:00:00 2001 From: Sebastian Ament Date: Mon, 23 Oct 2023 15:48:42 -0700 Subject: [PATCH] `NoisyExpectedHypervolumeMixin` (#1909) Summary: X-link: https://github.com/pytorch/botorch/pull/2045 Pull Request resolved: https://github.com/facebook/Ax/pull/1909 This commit introduces `NoisyExpectedHypervolumeMixin`, a derivative of `CachedCholeskyMCSamplerMixin` that separates out much of the Pareto-partitioning required for `qNEHVI`. Reviewed By: Balandat Differential Revision: D50337502 fbshipit-source-id: 1e24a4bc884ffa5b3fddb72730007e39fa3dae7e --- ax/models/tests/test_botorch_moo_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ax/models/tests/test_botorch_moo_model.py b/ax/models/tests/test_botorch_moo_model.py index d306596b5db..9e3b964de6c 100644 --- a/ax/models/tests/test_botorch_moo_model.py +++ b/ax/models/tests/test_botorch_moo_model.py @@ -10,6 +10,7 @@ from unittest import mock import ax.models.torch.botorch_moo_defaults as botorch_moo_defaults +import botorch.utils.multi_objective.hypervolume as hypervolume import numpy as np import torch from ax.core.search_space import SearchSpaceDigest @@ -49,7 +50,7 @@ "botorch.acquisition.factory.moo_monte_carlo.qExpectedHypervolumeImprovement" ) NEHVI_PARTITIONING_PATH = ( - "botorch.acquisition.multi_objective.monte_carlo.FastNondominatedPartitioning" + "botorch.utils.multi_objective.hypervolume.FastNondominatedPartitioning" ) EHVI_PARTITIONING_PATH = "botorch.acquisition.factory.FastNondominatedPartitioning" @@ -423,7 +424,7 @@ def test_BotorchMOOModel_with_qehvi( _mock_partitioning = es.enter_context( mock.patch( partitioning_path, - wraps=moo_monte_carlo.FastNondominatedPartitioning, + wraps=hypervolume.FastNondominatedPartitioning, ) ) torch_opt_config = TorchOptConfig(