code: def testLookup_1DInput(self, skip_gradient, timeout_ms):
      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,
          timeout_ms=timeout_ms,
      )
      self.assertAllClose(embedding.numpy(), [[1, 2]])
  
      embedding = de_ops.dynamic_embedding_lookup(
          ["first", "second", ""],
          self._config,
          "emb",
          service_address=self._kbs_address,
          skip_gradient_update=skip_gradient,
      )
      self.assertAllClose(embedding.numpy(), [[1, 2], [1, 2], [0, 0]])
  
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 3 for calling the DynamicEmbeddingOpsTest obj's underlying member method code with example usage: obj.testLookup_1DInput(skip_gradient, timeout_ms) without return types