From e458390678274b96ba56d43fbf6d1570a2f8afd1 Mon Sep 17 00:00:00 2001 From: who who who Date: Wed, 1 Mar 2023 12:33:56 +0800 Subject: [PATCH] fix wrong hints (#2817) --- tensorflow_addons/optimizers/average_wrapper.py | 4 +--- tensorflow_addons/optimizers/lookahead.py | 3 +-- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/tensorflow_addons/optimizers/average_wrapper.py b/tensorflow_addons/optimizers/average_wrapper.py index cfc4434cb9..2aab94927a 100644 --- a/tensorflow_addons/optimizers/average_wrapper.py +++ b/tensorflow_addons/optimizers/average_wrapper.py @@ -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 diff --git a/tensorflow_addons/optimizers/lookahead.py b/tensorflow_addons/optimizers/lookahead.py index c93d02c778..6e6b74e88e 100644 --- a/tensorflow_addons/optimizers/lookahead.py +++ b/tensorflow_addons/optimizers/lookahead.py @@ -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