remove deprecated assertDictContainsSubset

tensorflower-gardener · tensorflower-gardener · commit a23abb226bb7 · 2025-01-16T11:08:19.000-08:00
PiperOrigin-RevId: 716303100
diff --git a/tf_keras/optimizers/optimizer_test.py b/tf_keras/optimizers/optimizer_test.py
@@ -411,7 +411,7 @@ def get_config(self):
             "module": None,
             "registered_name": "CustomLRSchedule",
         }
-        self.assertDictContainsSubset(expected_config, config)
+        self.assertEqual(config, {**config, **expected_config})
         self.assertDictEqual(expected_learning_rate, config["learning_rate"])
 
         restored_optimizer = adam_new.Adam.from_config(

Original file line number	Diff line number	Diff line change
`@@ -411,7 +411,7 @@ def get_config(self):`
`411`	`411`	`"module": None,`
`412`	`412`	`"registered_name": "CustomLRSchedule",`
`413`	`413`	`}`
`414`		`- self.assertDictContainsSubset(expected_config, config)`
	`414`	`+ self.assertEqual(config, {config, expected_config})`
`415`	`415`	`self.assertDictEqual(expected_learning_rate, config["learning_rate"])`
`416`	`416`
`417`	`417`	`restored_optimizer = adam_new.Adam.from_config(`