Skip to content

Commit

Permalink
fix wrong hints (#2817)
Browse files Browse the repository at this point in the history
  • Loading branch information
fsx950223 authored Mar 1, 2023
1 parent 5876a83 commit e458390
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
4 changes: 1 addition & 3 deletions tensorflow_addons/optimizers/average_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,9 +45,7 @@ def __init__(

if not isinstance(optimizer, KerasLegacyOptimizer):
raise TypeError(
"optimizer is not an object of tf.keras.optimizers.Optimizer "
"or tf.keras.optimizers.legacy.Optimizer "
"(if you have tf version >= 2.11.0)."
"optimizer is not an object of tf.keras.optimizers.legacy.Optimizer "
)

self._optimizer = optimizer
Expand Down
3 changes: 1 addition & 2 deletions tensorflow_addons/optimizers/lookahead.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,8 +82,7 @@ def __init__(
optimizer = tf.keras.optimizers.get(optimizer)
if not isinstance(optimizer, KerasLegacyOptimizer):
raise TypeError(
"optimizer is not an object of tf.keras.optimizers.Optimizer "
"or tf.keras.optimizers.legacy.Optimizer (if you have tf version >= 2.11.0)."
"optimizer is not an object of tf.keras.optimizers.legacy.Optimizer "
)

self._optimizer = optimizer
Expand Down

0 comments on commit e458390

Please sign in to comment.