Re-enable memory leak test in TFP layers, since the underlying issue …

…in TF was fixed. PiperOrigin-RevId: 589239967
tensorflow · Dec 8, 2023 · fbc5ebe · fbc5ebe
1 parent abd1a1e
commit fbc5ebe
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py b/tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py
@@ -294,8 +294,6 @@ def testPropagatedAttributes(self):
 class MemoryLeakTest(test_util.TestCase):
 
   def testTypeObjectLeakage(self):
-    # TODO(b/303352281): Reenable this test.
-    self.skipTest('This test does not currently work under Python 3.11.')
 
     if not tf.executing_eagerly():
       self.skipTest('only relevant to eager')