code: def make_optimizer(self, **kwargs):
      if "learning_rate" not in kwargs:
          kwargs["learning_rate"] = 0.1
      if "momentum" not in kwargs:
          kwargs["momentum"] = 0.9
      if "use_nesterov" in kwargs:
          kwargs["nesterov"] = kwargs["use_nesterov"]
          del kwargs["use_nesterov"]
      if hasattr(tf.keras.optimizers, "legacy"):
          return optimizer_tests.WrappedTFOptimizer(
              tf.keras.optimizers.legacy.SGD(**kwargs)
          )
      return optimizer_tests.WrappedTFOptimizer(
          tf.keras.optimizers.SGD(**kwargs)
      )
  
imports: from sonnet.src import test_utils
  from sonnet.src.optimizers import momentum
  from sonnet.src.optimizers import optimizer_tests
  import tensorflow
label_desc: Use image node_id 1 for calling the ReferenceMomentumTest obj's underlying member method code with example usage: obj.make_optimizer() and returns: optimizer_tests, optimizer_tests