code: def testNoNestedPmap(self):
      devices = self._get_two_devices(require_same_type=True)
  
      def f(x):
          return x + 1.0
  
      data = tf_np.asarray(tf.convert_to_tensor(value=[1, 3]))
      with self.assertRaisesWithPredicateMatch(
          ValueError, r"Nested pmap is not supported"
      ):
          f = extensions.pmap(f, devices=devices)
          f = extensions.pmap(f, devices=devices)
          f(data)
  
imports: import functools
  import itertools
  from absl import flags
  from absl.testing import parameterized
  from jax import lax
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy import extensions
  import trax.tf_numpy.numpy
label_desc: Use image node_id 42 for calling the ExtensionsTest obj's underlying member method code with example usage: obj.testNoNestedPmap() and returns: unknown