diff --git a/xopt/generators/bayesian/bayesian_generator.py b/xopt/generators/bayesian/bayesian_generator.py index e1f2af45..e53057aa 100644 --- a/xopt/generators/bayesian/bayesian_generator.py +++ b/xopt/generators/bayesian/bayesian_generator.py @@ -418,10 +418,7 @@ def propose_candidates(self, model, n_candidates=1): ) else: candidates = self.numerical_optimizer.optimize( - acq_funct, - bounds, - n_candidates, - batch_initial_conditions=initial_points + acq_funct, bounds, n_candidates, batch_initial_conditions=initial_points ) return candidates diff --git a/xopt/tests/generators/bayesian/test_mobo.py b/xopt/tests/generators/bayesian/test_mobo.py index 26fd5b5d..fd8229ca 100644 --- a/xopt/tests/generators/bayesian/test_mobo.py +++ b/xopt/tests/generators/bayesian/test_mobo.py @@ -216,9 +216,9 @@ def test_initial_conditions(self): ic = gen._get_initial_conditions() assert not torch.allclose( ic[:4], - torch.tensor( - ((0.1, 0.1), (0.2, 0.2), (0.4, 0.2), (0.15, 0.15)) - ).reshape(4, 1, 2).double() + torch.tensor(((0.1, 0.1), (0.2, 0.2), (0.4, 0.2), (0.15, 0.15))) + .reshape(4, 1, 2) + .double(), ) gen.generate(1)