code: def simple_dlrm_model_config():
      """
      This simplified configuration is used during unittesting.
      """
  
      return DLRMModelConfig(
          dense_arch_layer_sizes=[32, 16, 8],
          dense_in_features=len(DEFAULT_INT_NAMES),
          embedding_dim=8,
          id_list_features_keys=DEFAULT_CAT_NAMES,
          num_embeddings_per_feature=len(DEFAULT_CAT_NAMES)
          * [
              3,
          ],
          over_arch_layer_sizes=[32, 32, 16, 1],
      )
  
imports: from dlrm_factory import DLRMFactory
  from dlrm_model_config import DLRMModelConfig
  from torchrec.datasets.criteo import DEFAULT_CAT_NAMES, DEFAULT_INT_NAMES
label_desc: Use image node_id 1 for calling a global function with example usage: simple_dlrm_model_config() and returns: DLRMModelConfig