code: def test_load_optimizer_state(self, tmpdir, shard_size):
      config_dict, hidden_dim, models = self._toy_model_config(
          shard_size
      )
      checkpoint_correctness_verification(
          config_dict,
          models,
          hidden_dim,
          tmpdir,
          load_optimizer_states=True,
      )
  
file_path: deepspeed/tests/unit/checkpoint/test_mics_optimizer.py
imports: import deepspeed
  from deepspeed.runtime.utils import required_torch_version
  from unit.common import DistributedTest
  from unit.simple_model import *
  from unit.checkpoint.common import *
  import pytest
label_desc: Use image node_id 2 for calling the TestMiCSCheckpoint obj's underlying member method code with example usage: obj.test_load_optimizer_state(tmpdir, shard_size) without return types