From 6351292b91827f8346c87609be77cea75ca0388c Mon Sep 17 00:00:00 2001 From: DevmatePyreFixMe Bot Date: Thu, 2 Jul 2026 09:56:12 -0700 Subject: [PATCH] fbcode/ax/generators/tests/test_utils.py Reviewed By: mgarrard Differential Revision: D110461339 --- ax/generators/tests/test_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ax/generators/tests/test_utils.py b/ax/generators/tests/test_utils.py index 4bf7e9882cf..ec05409a55e 100644 --- a/ax/generators/tests/test_utils.py +++ b/ax/generators/tests/test_utils.py @@ -64,7 +64,7 @@ def test_BestObservedPoint(self) -> None: ) X_obs = model.predict.mock_calls[0][1][0] self.assertEqual(X_obs.shape, (3, 2)) - # pyre-fixme[6]: For 2nd argument expected `Union[_SupportsArray[dtype[typing... + self.assertIsNotNone(xbest) self.assertTrue(np.array_equal(X_obs[1, :], xbest)) # 1 should be best # Test with specified utility baseline @@ -79,7 +79,7 @@ def test_BestObservedPoint(self) -> None: ) X_obs = model.predict.mock_calls[1][1][0] self.assertEqual(X_obs.shape, (3, 2)) - # pyre-fixme[6]: For 2nd argument expected `Union[_SupportsArray[dtype[typing... + self.assertIsNotNone(xbest) self.assertTrue(np.array_equal(X_obs[2, :], xbest)) # 2 should be best # Test with feasibility threshold @@ -94,7 +94,7 @@ def test_BestObservedPoint(self) -> None: ) X_obs = model.predict.mock_calls[2][1][0] self.assertEqual(X_obs.shape, (3, 2)) - # pyre-fixme[6]: For 2nd argument expected `Union[_SupportsArray[dtype[typing... + self.assertIsNotNone(xbest) self.assertTrue(np.array_equal(X_obs[0, :], xbest)) # 0 should be best # Parameter infeasible