code: def setUp(self):
      super().setUp()
  
      # TODO(b/261485237): Enable CPU testing once CollectivePermute is supported
      # on CPU's.
      if not test_util.is_tpu_present():
          self.skipTest("This test only runs on TPUs.")
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python import numpy_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager.polymorphic_function import polymorphic_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import special_math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the SpatiallyPartitionedConvOpTest obj's underlying member method code with example usage: obj.setUp() without return types