code: def _make_distillation_spec(type_):
      output = distillation_spec_pb2.DistillationSpec()
      output.distillation_type = type_
      output.balance_losses_spec.num_ramp_trials = 100
      return output
  
file_path: model_search/model_search/meta/distillation_test.py
imports: from absl.testing import parameterized
  from model_search.meta import distillation
  from model_search.proto import distillation_spec_pb2
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 1 for calling a global function with example usage: _make_distillation_spec(type_) and returns: output