code: def setUpClass(cls):
      super().setUpClass()
      cluster_def = multi_worker_test_base.create_in_process_cluster(
          num_workers=2, num_ps=2
      )
      cls.cluster_resolver = (
          tf.distribute.cluster_resolver.SimpleClusterResolver(
              tf.train.ClusterSpec(cluster_def)
          )
      )
  
imports: import os
  from absl.testing import parameterized
  import tensorflow
  import tensorflow.compat.v1
  from tensorflow.python.distribute import combinations
  from tensorflow.python.distribute import multi_worker_test_base
  from tensorflow.python.distribute import parameter_server_strategy_v2
  from tensorflow.python.distribute import sharded_variable
  from tensorflow.python.distribute import strategy_combinations
  from tensorflow.python.distribute import test_util
  from tensorflow.python.distribute import values
  from tensorflow.python.eager import context
  from tensorflow.python.eager import test
  from tensorflow.python.ops import lookup_ops
label_desc: Use image node_id 1 for calling the PSStrategySaveAndLoadTest obj's underlying member method code with example usage: obj.setUpClass(cls) without return types