code: def testEvalOnShapesNested(self):
      transformer = functools.partial(
          extensions.eval_on_shapes, allow_static_outputs=True
      )
  
      @transformer
      def outer():
          @transformer
          def inner():
              return 1
  
          return inner() + 2
  
      r = outer()
      self.assertIsInstance(r, int)
      self.assertEqual(3, r)
  
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 13 for calling the ExtensionsTest obj's underlying member method code with example usage: obj.testEvalOnShapesNested() and returns: unknown, int