pylint

secondmind-labs · May 15, 2024 · 1cf1916 · 1cf1916
1 parent 2450484
commit 1cf1916
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/unit/test_ask_tell_optimization.py b/tests/unit/test_ask_tell_optimization.py
@@ -44,7 +44,7 @@
 from trieste.models.interfaces import ProbabilisticModel, TrainableProbabilisticModel
 from trieste.objectives.utils import mk_batch_observer
 from trieste.observer import OBJECTIVE
-from trieste.space import Box, SearchSpace
+from trieste.space import Box
 from trieste.types import State, Tag, TensorType
 from trieste.utils.misc import LocalizedTag
 

diff --git a/tests/unit/test_bayesian_optimizer.py b/tests/unit/test_bayesian_optimizer.py
@@ -26,11 +26,11 @@
 from tests.unit.test_ask_tell_optimization import DatasetChecker
 from tests.util.misc import (
     FixedAcquisitionRule,
+    FixedLocalAcquisitionRule,
     assert_datasets_allclose,
     empty_dataset,
     mk_dataset,
     quadratic,
-    FixedLocalAcquisitionRule,
 )
 from tests.util.models.gpflow.models import (
     GaussianProcess,