From 6974a9b0059c581132622e0d5e049939dfa38f3d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Yan=20Facai=20=28=E9=A2=9C=E5=8F=91=E6=89=8D=29?= Date: Wed, 22 May 2019 20:57:33 +0800 Subject: [PATCH] TST: fix broken adam test --- tensorflow_addons/optimizers/lazy_adam_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_addons/optimizers/lazy_adam_test.py b/tensorflow_addons/optimizers/lazy_adam_test.py index 9a3adab71d..c0fd2bf311 100644 --- a/tensorflow_addons/optimizers/lazy_adam_test.py +++ b/tensorflow_addons/optimizers/lazy_adam_test.py @@ -212,7 +212,7 @@ def doTestBasic(self, use_callable_params=False): self.evaluate(var0)) self.assertAllCloseAccordingToType(var1_np, self.evaluate(var1)) - self.assertEqual("var0_%d/m:0" % (i,), + self.assertEqual("Adam/var0_%d/m:0" % (i,), opt.get_slot(var0, "m").name) @test_utils.run_in_graph_and_eager_modes(reset_test=True)