mypy

secondmind-labs · Jan 20, 2025 · 3c7ec1a · 3c7ec1a
1 parent 5d32c57
commit 3c7ec1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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