Fix merge issues

tests/unit/acquisition/test_rule.py

@@ -23,7 +23,6 @@
 import pytest
 import tensorflow as tf
 import tensorflow_probability as tfp
-import tensorflow_probability as tfp
 
 from tests.util.misc import empty_dataset, mk_dataset, quadratic, random_seed
 from tests.util.models.gpflow.models import (

tests/unit/test_ask_tell_optimization.py

@@ -15,7 +15,6 @@
 
 from typing import Mapping, Optional
 
-import numpy.testing as npt
 import numpy.testing as npt
 import pytest
 import tensorflow as tf

trieste/acquisition/rule.py

@@ -20,7 +20,6 @@
 import copy
 from abc import ABC, abstractmethod
 from collections import Counter
-from collections import Counter
 from collections.abc import Mapping
 from dataclasses import dataclass
 from typing import (
@@ -36,19 +35,6 @@
     cast,
     overload,
 )
-from typing import (
-    Any,
-    Callable,
-    Generic,
-    Optional,
-    Sequence,
-    Set,
-    Tuple,
-    TypeVar,
-    Union,
-    cast,
-    overload,
-)
 
 import numpy as np
 from check_shapes import check_shapes, inherit_check_shapes

trieste/acquisition/utils.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import copy
-import copy
 import functools
 from typing import Dict, Mapping, Tuple, Union
 
@@ -22,8 +21,6 @@
 from ..data import Dataset
 from ..models import ProbabilisticModelType
 from ..observer import OBJECTIVE
-from ..models import ProbabilisticModelType
-from ..observer import OBJECTIVE
 from ..space import SearchSpaceType
 from ..types import Tag, TensorType
 from ..utils.misc import LocalizedTag

trieste/ask_tell_optimization.py

@@ -53,7 +53,6 @@
 from .types import State, Tag, TensorType
 from .utils import Ok, Timer
 from .utils.misc import LocalizedTag, get_value_for_tag, ignoring_local_tags
-from .utils.misc import LocalizedTag, get_value_for_tag, ignoring_local_tags
 
 StateType = TypeVar("StateType")
 """ Unbound type variable. """

trieste/bayesian_optimizer.py

@@ -63,13 +63,11 @@
 from .data import Dataset
 from .models import SupportsCovarianceWithTopFidelity, TrainableProbabilisticModel
 from .objectives.utils import mk_batch_observer
-from .objectives.utils import mk_batch_observer
 from .observer import OBJECTIVE, Observer
 from .space import SearchSpace
 from .types import State, Tag, TensorType
 from .utils import Err, Ok, Result, Timer
 from .utils.misc import LocalizedTag, get_value_for_tag, ignoring_local_tags
-from .utils.misc import LocalizedTag, get_value_for_tag, ignoring_local_tags
 
 StateType = TypeVar("StateType")
 """ Unbound type variable. """

trieste/experimental/plotting/plotting.py

@@ -34,12 +34,10 @@
 from trieste.acquisition.multi_objective.dominance import non_dominated
 from trieste.bayesian_optimizer import FrozenRecord, Record, StateType
 from trieste.observer import OBJECTIVE
-from trieste.observer import OBJECTIVE
 from trieste.space import TaggedMultiSearchSpace
 from trieste.types import TensorType
 from trieste.utils import to_numpy
 from trieste.utils.misc import LocalizedTag
-from trieste.utils.misc import LocalizedTag
 
 
 def create_grid(