code: def setUp(self):
      super(TPUEstimatorTest, self).setUp()
  
      if not tf_compat.version_greater_or_equal("1.14.0"):
          self.skipTest(
              "TPUEmbedding not supported in version 1.13.0 and below."
          )
  
      # TPUConfig initializes model_dir from TF_CONFIG and checks that the user
      # provided model_dir matches the TF_CONFIG one.
      tf_config = {"model_dir": self.test_subdirectory}
      os.environ["TF_CONFIG"] = json.dumps(tf_config)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import itertools
  import json
  import os
  from absl.testing import parameterized
  from adanet import tf_compat
  from adanet.core import testing_utils
  from adanet.core.tpu_estimator import TPUEstimator
  from adanet.subnetwork import Builder
  from adanet.subnetwork import Report
  from adanet.subnetwork import SimpleGenerator
  from adanet.subnetwork import Subnetwork
  import numpy
  import tensorflow.compat.v2
  from tensorflow_estimator.python.estimator.head import regression_head
label_desc: Use image node_id 1 for calling the TPUEstimatorTest obj's underlying member method code with example usage: obj.setUp() without return types