Skip to content

Commit

Permalink
review of acq. tests & docs
Browse files Browse the repository at this point in the history
  • Loading branch information
cahity committed Nov 4, 2024
1 parent 83aad25 commit e3d8373
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions test/acquisition/test_acquisition.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,6 @@ class TestMaxDiagonalAcquisition(TestCase):

def test_forward(self):
"""Test the forward method."""
q = 2
choices = np.array([[1, 2], [3, 4], [5, 6]])
design_space = FixedPointsDesignSpace(choices, 2)
mock_model = mock.MagicMock()
Expand All @@ -99,7 +98,6 @@ def test_forward(self):
design_space.update(mock_model, np.array([1]), [0, 1, 2])
axq = MaxDiagonalAcquisition(design_space)
ret_values = axq(choices)
print(np.sqrt(2) * np.array([2, 3, 4]))
self.assertTrue(np.allclose(ret_values, np.sqrt(2) * np.array([2, 2 * np.sqrt(3), 4])))


Expand Down

0 comments on commit e3d8373

Please sign in to comment.