From fbc5ebe9b1d343113fb917010096cfd88b32eecf Mon Sep 17 00:00:00 2001 From: emilyaf Date: Fri, 8 Dec 2023 14:16:46 -0800 Subject: [PATCH] Re-enable memory leak test in TFP layers, since the underlying issue in TF was fixed. PiperOrigin-RevId: 589239967 --- .../layers/internal/distribution_tensor_coercible_test.py | 2 -- 1 file changed, 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 index e84d69d642..85bf8380f7 100644 --- 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')