diff --git a/trieste/models/keras/architectures.py b/trieste/models/keras/architectures.py index ea2619b59..317928d71 100644 --- a/trieste/models/keras/architectures.py +++ b/trieste/models/keras/architectures.py @@ -33,7 +33,7 @@ except AttributeError: # pragma: no cover (tested but not by coverage) try: # in TF 2.14 it's in keras but not in tf_keras (but vice versa in TF 2.16)! - from keras.src.saving.serialization_lib import SafeModeScope + from keras.src.saving.serialization_lib import SafeModeScope # type:ignore[no-redef] except (ImportError, ModuleNotFoundError): SafeModeScope = contextlib.nullcontext from tensorflow_probability.python.layers.distribution_layer import DistributionLambda, _serialize