code: def testUpdate_1DInput(self, use_kbs_address, skip_gradient):
      init = self._config.knowledge_bank_config.initializer
      init.default_embedding.value.append(1)
      init.default_embedding.value.append(2)
      embedding = de_ops.dynamic_embedding_lookup(
          ["first"],
          self._config,
          "emb",
          service_address=self._kbs_address,
          skip_gradient_update=skip_gradient,
      )
      self.assertAllClose(embedding.numpy(), [[1, 2]])
  
      update_res = de_ops.dynamic_embedding_update(
          ["first"],
          tf.constant([[2.0, 4.0]]),
          self._config,
          "emb",
          service_address=self._kbs_address,
      )
      self.assertAllClose(update_res.numpy(), [[2, 4]])
  
      embedding = de_ops.dynamic_embedding_lookup(
          ["first"],
          self._config,
          "emb",
          service_address=self._kbs_address,
          skip_gradient_update=skip_gradient,
      )
      self.assertAllClose(embedding.numpy(), [[2, 4]])
  
      # Allows keys' shape to be [N, 1] and values shape to be [N, D].
      update_res = de_ops.dynamic_embedding_update(
          [["first"]],
          tf.constant([[4.0, 5.0]]),
          self._config,
          "emb",
          service_address=self._kbs_address,
      )
      self.assertAllClose(update_res.numpy(), [[4, 5]])
      embedding = de_ops.dynamic_embedding_lookup(
          ["first"],
          self._config,
          "emb",
          service_address=self._kbs_address,
          skip_gradient_update=skip_gradient,
      )
      self.assertAllClose(embedding.numpy(), [[4, 5]])
  
imports: import itertools
  from absl.testing import parameterized
  from research.carls import context
  from research.carls import dynamic_embedding_ops
  from research.carls.testing import test_util
  import numpy
  import tensorflow
label_desc: Use image node_id 5 for calling the DynamicEmbeddingOpsTest obj's underlying member method code with example usage: obj.testUpdate_1DInput(use_kbs_address, skip_gradient) without return types