Skip to content

Commit

Permalink
Merge f1387a5 into 6268ced
Browse files Browse the repository at this point in the history
  • Loading branch information
lena-kashtelyan authored Apr 11, 2024
2 parents 6268ced + f1387a5 commit 54b0fac
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions ax/models/torch/cbo_lcea.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@

from ax.core.search_space import SearchSpaceDigest
from ax.core.types import TCandidateMetadata
from ax.models.torch.alebo import ei_or_nei
from ax.models.torch.botorch import BotorchModel
from ax.models.torch.botorch_defaults import get_qLogNEI
from ax.models.torch.cbo_sac import generate_model_space_decomposition
from ax.models.torch_base import TorchModel, TorchOptConfig
from ax.utils.common.docutils import copy_doc
Expand Down Expand Up @@ -111,8 +111,7 @@ def __init__(
# pyre-fixme[4]: Attribute must be annotated.
self.train_embedding = self.gp_model_args.get("train_embedding", True)
super().__init__(
model_constructor=self.get_and_fit_model,
acqf_constructor=ei_or_nei, # pyre-ignore
model_constructor=self.get_and_fit_model, acqf_constructor=get_qLogNEI
)

@copy_doc(TorchModel.fit)
Expand Down

0 comments on commit 54b0fac

Please sign in to comment.