Skip to content

Commit 6f412b4

Browse files
yileitensorflower-gardener
authored andcommitted
Re-enable these tests now that the TF memory leak is fixed.
PiperOrigin-RevId: 592578907
1 parent f458edc commit 6f412b4

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

tf_keras/layers/kernelized_test.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,6 @@ def test_call(self, initializer, trainable):
150150

151151
@tf_test_utils.assert_no_new_pyobjects_executing_eagerly()
152152
def test_no_eager_Leak(self):
153-
self.skipTest("b/301158510")
154153
# Tests that repeatedly constructing and building a Layer does not leak
155154
# Python objects.
156155
inputs = tf.random.uniform((5, 4), seed=1)

tf_keras/legacy_tf_layers/core_test.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,6 @@ def testCall(self):
9393

9494
@tf_test_utils.assert_no_new_pyobjects_executing_eagerly()
9595
def testNoEagerLeak(self):
96-
self.skipTest("b/301158510")
9796
# Tests that repeatedly constructing and building a Layer does not leak
9897
# Python objects.
9998
inputs = tf.random.uniform((5, 4), seed=1)

0 commit comments

Comments
 (0)