docstr: Test ModuleSummary callback with train entry point
code: def test_module_summary_max_depth(self) -> None:
      """
      Test ModuleSummary callback with train entry point
      """
      input_dim = 2
      dataset_len = 10
      batch_size = 2
      max_epochs = 1
  
      dataloader = generate_random_dataloader(
          dataset_len, input_dim, batch_size
      )
      state = State(
          entry_point=EntryPoint.TRAIN,
          train_state=PhaseState(
              dataloader=dataloader,
              max_epochs=max_epochs,
          ),
      )
  
      my_unit = MagicMock(spec=DummyTrainUnit)
      module_summary_callback = ModuleSummary(max_depth=2)
      module_summary_callback.on_train_epoch_start(state, my_unit)
      self.assertEqual(module_summary_callback._max_depth, 2)
  
file_path: tnt/tests/framework/callbacks/test_module_summary.py
imports: import unittest
  from unittest.mock import MagicMock
  import torch
  from torchtnt.framework._test_utils import DummyAutoUnit, DummyTrainUnit, generate_random_dataloader
  from torchtnt.framework.callbacks.module_summary import ModuleSummary
  from torchtnt.framework.state import EntryPoint, PhaseState, State
  from torchtnt.utils.version import is_torch_version_geq_1_13
label_desc: Use image node_id 1 for calling the ModuleSummaryTest obj's underlying member method code with example usage: obj.test_module_summary_max_depth() without return types