code: def setUp(self) -> None:
      self.tables = [
          EmbeddingConfig(
              name="table_0",
              feature_names=["feature_0"],
              embedding_dim=4,
              num_embeddings=4,
          ),
          EmbeddingConfig(
              name="table_1",
              feature_names=["feature_1"],
              embedding_dim=4,
              num_embeddings=4,
          ),
      ]
  
      self.constraints = {
          "table_0": ParameterConstraints(
              cache_params=CacheParams(
                  algorithm=CacheAlgorithm.LRU,
                  load_factor=0.1,
                  reserved_memory=8.0,
                  precision=DataType.FP16,
              ),
              enforce_hbm=True,
              stochastic_rounding=False,
              bounds_check_mode=BoundsCheckMode.IGNORE,
          ),
          "table_1": ParameterConstraints(
              cache_params=CacheParams(
                  algorithm=CacheAlgorithm.LFU,
                  load_factor=0.2,
                  reserved_memory=0.0,
                  precision=DataType.FP16,
              ),
              enforce_hbm=True,
              stochastic_rounding=False,
              bounds_check_mode=BoundsCheckMode.NONE,
          ),
      }
  
      self.model = EmbeddingCollection(tables=self.tables)
      self.sharder = EmbeddingCollectionSharder()
      planner = EmbeddingShardingPlanner(
          topology=Topology(world_size=1, compute_device="cpu"),
          constraints=self.constraints,
      )
      self.expected_plan = planner.plan(
          self.model, [self.sharder]
      )  # pyre-ignore[6]
  
      self.expected_sharding_infos = create_sharding_infos_by_sharding(
          self.model,
          self.expected_plan.get_plan_for_module(""),  # pyre-ignore[6]
          fused_params=None,
      )
  
file_path: torchrec/torchrec/distributed/planner/tests/test_embedding_utils.py
imports: import copy
  import unittest
  from torchrec.distributed.embedding import create_sharding_infos_by_sharding, EmbeddingCollectionSharder
  from torchrec.distributed.planner import EmbeddingShardingPlanner, ParameterConstraints, Topology
  from torchrec.distributed.types import BoundsCheckMode, CacheAlgorithm, CacheParams, DataType
  from torchrec.modules.embedding_configs import EmbeddingConfig
  from torchrec.modules.embedding_modules import EmbeddingCollection
label_desc: Use image node_id 1 for calling the CreateShardingInfoTest obj's underlying member method code with example usage: obj.setUp() without return types