code: def test_tf_should_error_with_more_than_one_named_axis(self):
      if JAX_MODE:
          self.skipTest("Test only applies to TF backend.")
      with self.assertRaisesRegex(
          ValueError,
          "TensorFlow backend does not support multiple shard axes",
      ):
          distribute_lib.canonicalize_named_axis(["a", "b"])
  
imports: import functools
  import itertools
  from absl.testing import parameterized
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import bernoulli
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import distribute_lib
  from tensorflow_probability.python.internal import distribute_test_lib
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 1 for calling the CollectiveTest obj's underlying member method code with example usage: obj.test_tf_should_error_with_more_than_one_named_axis() without return types