Re-enable these tests now that the TF memory leak is fixed.

PiperOrigin-RevId: 592578907
keras-team · Dec 20, 2023 · 6f412b4 · 6f412b4
1 parent f458edc
commit 6f412b4
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/tf_keras/layers/kernelized_test.py b/tf_keras/layers/kernelized_test.py
@@ -150,7 +150,6 @@ def test_call(self, initializer, trainable):
 
     @tf_test_utils.assert_no_new_pyobjects_executing_eagerly()
     def test_no_eager_Leak(self):
-        self.skipTest("b/301158510")
         # Tests that repeatedly constructing and building a Layer does not leak
         # Python objects.
         inputs = tf.random.uniform((5, 4), seed=1)

diff --git a/tf_keras/legacy_tf_layers/core_test.py b/tf_keras/legacy_tf_layers/core_test.py
@@ -93,7 +93,6 @@ def testCall(self):
 
     @tf_test_utils.assert_no_new_pyobjects_executing_eagerly()
     def testNoEagerLeak(self):
-        self.skipTest("b/301158510")
         # Tests that repeatedly constructing and building a Layer does not leak
         # Python objects.
         inputs = tf.random.uniform((5, 4), seed=1)