code: def __init__(self, ebc: QuantEmbeddingBagCollection) -> None:
      super().__init__()
      self.ebc = ebc
  
imports: import unittest
  from typing import List
  import torch
  from hypothesis import given, settings, strategies, Verbosity
  from torch import nn
  from torchrec.distributed.model_parallel import DistributedModelParallel
  from torchrec.distributed.types import ShardingEnv
  from torchrec.modules.embedding_configs import DataType, EmbeddingBagConfig
  from torchrec.modules.embedding_modules import EmbeddingBagCollection
  from torchrec.quant.embedding_modules import EmbeddingBagCollection
  from torchrec.quant.utils import meta_to_cpu_placement
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor, KeyedTensor
label_desc: Use image node_id 1 to create a new TestSparseArch object from inherited base classes: nn with example: obj = TestSparseArch(ebc)