code: def setUp(self):
      super().setUp()
      self.ckpt_dir = self.create_tempdir()
      self.w0 = np.array([[2.0, -1.0]])
      self.w1 = np.array([[3.0, -2.0]])
      model = self._build_toy_model()
      ckpt = tf.train.Checkpoint(model=model)
      model.weights[0].assign(self.w0)
      ckpt.save(os.path.join(self.ckpt_dir, "ckpt"))  # ckpt-1
      model.weights[0].assign(self.w1)
      ckpt.save(os.path.join(self.ckpt_dir, "ckpt"))
  
file_path: neural-structured-learning/research/multi_representation_adversary/multi_representation_adversary/evaluator_test.py
imports: import os
  from absl.testing import parameterized
  import gin
  from multi_representation_adversary import evaluator
  from multi_representation_adversary import resnet
  import numpy
  import tensorflow.compat.v2
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling the AggregationTest obj's underlying member method code with example usage: obj.setUp() without return types