From be84b38063a1362f9612487e50105c4f8af7a235 Mon Sep 17 00:00:00 2001 From: jburnim Date: Fri, 3 Nov 2023 12:21:42 -0700 Subject: [PATCH] Update one more tf1.layers reference to tf_keras.tf1_layers. PiperOrigin-RevId: 579266654 --- .../python/bijectors/batch_normalization_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_probability/python/bijectors/batch_normalization_test.py b/tensorflow_probability/python/bijectors/batch_normalization_test.py index bb29345223..7a1604380d 100644 --- a/tensorflow_probability/python/bijectors/batch_normalization_test.py +++ b/tensorflow_probability/python/bijectors/batch_normalization_test.py @@ -220,7 +220,7 @@ def testInvertMutuallyConsistent(self, layer_cls): def testWithKeras(self): # NOTE: Keras throws an error below if we use - # tf1.layers.BatchNormalization() here. + # tf_keras.tf1_layers.BatchNormalization() here. layer = None dist = transformed_distribution.TransformedDistribution(