code: def testTransform(self):
      configs = [
          EmbeddingConfig(name="A", num_embeddings=8, embedding_dim=32),
          EmbeddingConfig(name="B", num_embeddings=8, embedding_dim=32),
      ]
      transformer_collection = IDTransformerCollection(configs)
      global_kjt = KeyedJaggedTensor(
          keys=["B", "A"],
          values=torch.tensor([1, 3, 2, 3, 4, 3, 2]),
          lengths=torch.tensor([4, 3]),
      )
      cache_kjt, fetch_handles = transformer_collection.transform(
          global_kjt
      )
      for handle in fetch_handles:
          handle.wait()
      self.assertEqual(cache_kjt.keys(), global_kjt.keys())
      self.assertTrue(
          torch.all(cache_kjt.lengths() == global_kjt.lengths())
      )
      self.assertTrue(
          torch.all(
              cache_kjt.values() == torch.tensor([0, 1, 2, 1, 0, 1, 2])
          )
      )
  
imports: import unittest
  import torch
  from torchrec import EmbeddingBagConfig, EmbeddingConfig, KeyedJaggedTensor
  from torchrec_dynamic_embedding.id_transformer_collection import IDTransformerCollection
label_desc: Use image node_id 1 for calling the TestIDTransformerCollection obj's underlying member method code with example usage: obj.testTransform() without return types