code: def test_new_class_declared_nesting_init(self):
      ds_config = dict(
          train_batch_size=1, zero_optimization=dict(stage=3)
      )
  
      with deepspeed.zero.Init(config_dict_or_path=ds_config):
  
          class MyModel(torch.nn.Module):
              def __init__(self):
                  super().__init__()
                  self.fc = torch.nn.Linear(4, 4)
  
          with deepspeed.zero.Init(config_dict_or_path=ds_config):
              model = MyModel()
  
      # ensure that zero3 processed the parameter
      assert hasattr(model.fc.weight, "ds_id")
      deepspeed_engine, *_ = deepspeed.initialize(
          model=model, config_params=ds_config
      )
  
file_path: DeepSpeed/tests/unit/runtime/zero/test_zero_dynamic_class.py
imports: import torch
  from unit.common import DistributedTest
  import deepspeed
label_desc: Use image node_id 1 for calling the TestNewClassDeclaredNestingInit obj's underlying member method code with example usage: obj.test_new_class_declared_nesting_init() without return types