code: def test_load_and_aggregate_avg_weight(self):
      model = evaluator.load_and_aggregate(
          self.ckpt_dir, [1, 2], "avg_weight", self._build_toy_model
      )
      self.assertAllClose(
          (self.w0 + self.w1) / 2, model(np.array([[1.0]]))
      )
  
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 3 for calling the AggregationTest obj's underlying member method code with example usage: obj.test_load_and_aggregate_avg_weight() without return types