small fix

tensorflow · May 19, 2022 · 0bfa0d0 · 0bfa0d0
1 parent a2a3e27
commit 0bfa0d0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tensorflow_addons/optimizers/average_wrapper.py b/tensorflow_addons/optimizers/average_wrapper.py
@@ -39,7 +39,8 @@ def __init__(
             (tf.keras.optimizers.legacy.Optimizer, tf.keras.optimizers.Optimizer),
         ):
             raise TypeError(
-                "optimizer is not an object of tf.keras.optimizers.legacy.Optimizer"
+                "optimizer is not an object of tf.keras.optimizers.legacy.Optimizer "
+                "or tf.keras.optimizers.Optimizer."
             )
 
         self._optimizer = optimizer

diff --git a/tensorflow_addons/optimizers/lookahead.py b/tensorflow_addons/optimizers/lookahead.py
@@ -76,7 +76,8 @@ def __init__(
             (tf.keras.optimizers.legacy.Optimizer, tf.keras.optimizers.Optimizer),
         ):
             raise TypeError(
-                "optimizer is not an object of tf.keras.optimizers.legacy.Optimizer"
+                "optimizer is not an object of tf.keras.optimizers.legacy.Optimizer "
+                "or tf.keras.optimizers.Optimizer."
             )
 
         self._optimizer = optimizer