fix linting

xopt-org · Oct 29, 2024 · 7f1311d · 7f1311d
1 parent d55eedf
commit 7f1311d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git 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
@@ -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)