code: def test_embedding(self, optimizer_name):
      if optimizer_name != "sgd":
          self.skip_if_oss()
      self._test_embedding(
          optimizer_name,
          training=True,
          sparse=True,
          is_high_dimensional=True,
      )
  
imports: from absl.testing import parameterized
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.platform import test
  from tensorflow.python.tpu.tests import tpu_embedding_v2_correctness_base_test
label_desc: Use image node_id 1 for calling the TPUEmbeddingCorrectnessTest obj's underlying member method code with example usage: obj.test_embedding(optimizer_name) without return types