Skip to content

Commit d6d8591

Browse files
liangshi7facebook-github-bot
authored andcommitted
remove fidelity_dims member in botorch MF MVE class (#336)
Summary: Pull Request resolved: #336 the learned cost model does not have fidelity_dims member. Since it is not used at all, remove it to avoid errors. Reviewed By: bkarrer Differential Revision: D18825344 fbshipit-source-id: 2cd187b93b44bda32423e799c0718351bc39aa25
1 parent 77addf5 commit d6d8591

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

botorch/acquisition/max_value_entropy_search.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,6 @@ def __init__(
378378
if cost_aware_utility is None:
379379
cost_model = AffineFidelityCostModel(fidelity_weights={-1: 1.0})
380380
cost_aware_utility = InverseCostWeightedUtility(cost_model=cost_model)
381-
self.fidelity_dims = cost_aware_utility.cost_model.fidelity_dims
382381

383382
self.cost_aware_utility = cost_aware_utility
384383
self.expand = expand

test/acquisition/test_max_value_entropy_search.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,6 @@ def test_q_multi_fidelity_max_value_entropy(self):
130130
self.assertIsInstance(qMF_MVE.expand, Callable)
131131
self.assertIsInstance(qMF_MVE.project, Callable)
132132
self.assertIsNone(qMF_MVE.X_pending)
133-
self.assertEqual(qMF_MVE.fidelity_dims, [-1])
134133
self.assertEqual(qMF_MVE.posterior_max_values.shape, torch.Size([10, 1]))
135134

136135
# test evaluation

0 commit comments

Comments
 (0)