Skip to content

Commit 70c9b27

Browse files
committed
Reverted changes
1 parent bcbb694 commit 70c9b27

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

tensorflow_addons/optimizers/conditional_gradient.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,6 @@ def __init__(
8686
compatibility, recommended to use `learning_rate` instead.
8787
"""
8888
super().__init__(name=name, **kwargs)
89-
self.__name__ = name
9089
self._set_hyper("learning_rate", kwargs.get("lr", learning_rate))
9190
self._set_hyper("lambda_", lambda_)
9291
self.epsilon = epsilon or tf.keras.backend.epsilon()

tensorflow_addons/optimizers/tests/conditional_gradient_test.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
import tensorflow as tf
2222
from tensorflow_addons.utils import test_utils
2323
from tensorflow_addons.optimizers import conditional_gradient as cg_lib
24-
from tensorflow_addons.testing import serialization
2524

2625

2726
def _dtypes_to_test(use_gpu):
@@ -1469,4 +1468,4 @@ def test_serialization():
14691468
)
14701469
config = tf.keras.optimizers.serialize(optimizer)
14711470
new_optimizer = tf.keras.optimizers.deserialize(config)
1472-
serialization.check_config(new_optimizer.get_config(), optimizer, False)
1471+
assert optimizer.get_config() == new_optimizer.get_config()

tensorflow_addons/testing/serialization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def check_config(config, class_, strict):
7676
class_.__name__
7777
)
7878
)
79-
if parameter_name != "kwargs" and parameter_name not in config:
79+
if parameter_name not in config:
8080
raise KeyError(
8181
"The constructor parameter {} is not present in the config dict "
8282
"obtained with `.get_config()` of {}. All parameters should be set to "

0 commit comments

Comments
 (0)