Update one more tf1.layers reference to tf_keras.tf1_layers.

PiperOrigin-RevId: 579266654
tensorflow · Nov 3, 2023 · be84b38 · be84b38
1 parent 65d9d72
commit be84b38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(