code: def is_leaf_module(
      self, m: torch.nn.Module, module_qualified_name: str
  ) -> bool:
      if (
          type(m).__name__ in self._leaf_module_names
          or module_qualified_name in self._leaf_module_names
      ):
          return True
      return super().is_leaf_module(m, module_qualified_name)
  
file_path: torchrec/torchrec/modules/tests/test_mc_embedding_modules.py
imports: import unittest
  from copy import deepcopy
  from typing import cast, Dict, List, Optional
  import torch
  from torchrec.modules.embedding_configs import EmbeddingBagConfig, EmbeddingConfig
  from torchrec.modules.embedding_modules import EmbeddingBagCollection, EmbeddingCollection
  from torchrec.modules.mc_embedding_modules import ManagedCollisionEmbeddingBagCollection, ManagedCollisionEmbeddingCollection
  from torchrec.modules.mc_modules import DistanceLFU_EvictionPolicy, ManagedCollisionCollection, ManagedCollisionModule, MCHManagedCollisionModule
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor, KeyedTensor
label_desc: Use image node_id 2 for calling the Tracer obj's underlying member method code with example usage: obj.is_leaf_module(m, module_qualified_name) and returns: super, True