code: def setUp(self):
      super().setUp()
      test_util.reset_logical_devices("CPU", 2)
      if test_util.is_gpu_present():
          test_util.reset_logical_devices("GPU", 2)
  
file_path: tensorflow/tensorflow/dtensor/python/tests/config_test.py
imports: import os
  from tensorflow.dtensor.python import config
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.eager import context
  from tensorflow.python.framework import device
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the ConfigTest obj's underlying member method code with example usage: obj.setUp() without return types