code: def testIncompatibleShapes(self):
      bn = tf.keras.layers.BatchNormalization(axis=-1)
      layer = squashed_outer_wrapper.SquashedOuterWrapper(
          bn, inner_rank=3
      )
  
      with self.assertRaisesRegex(ValueError, "must have known rank"):
          fn = common.function(layer)
          fn.get_concrete_function(
              tf.TensorSpec(shape=None, dtype=tf.float32)
          )
  
file_path: agents/tf_agents/keras_layers/squashed_outer_wrapper_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tf_agents.keras_layers import squashed_outer_wrapper
  from tf_agents.utils import common
  from tf_agents.utils import test_utils
label_desc: Use image node_id 3 for calling the SquashedOuterWrapperTest obj's underlying member method code with example usage: obj.testIncompatibleShapes() without return types