code: def test_not_load_optimizer_state(
      self, tmpdir, zero_stage, use_cpu_offload, adam_optimizer
  ):
      if (
          use_cpu_offload
          and not deepspeed.ops.__compatible_ops__[CPUAdamBuilder.NAME]
      ):
          pytest.skip("cpu-adam is not compatible")
  
      config_dict = {
          "train_batch_size": 2,
          "steps_per_print": 1,
          "optimizer": {
              "type": "Adam",
              "params": {
                  "lr": 0.00015,
                  "betas": [0.8, 0.999],
                  "eps": 1e-8,
                  "weight_decay": 3e-7,
              },
          },
          "fp16": {"enabled": True},
          "zero_optimization": {
              "stage": zero_stage,
              "cpu_offload": use_cpu_offload,
          },
      }
      hidden_dim = 10
  
      if zero_stage == 3:
          global DeepSpeedZeroOptimizer_Stage3
          from deepspeed.runtime.zero.stage3 import (
              DeepSpeedZeroOptimizer_Stage3,
          )
  
          with deepspeed.zero.Init():
              models = [
                  SimpleModel(hidden_dim, empty_grad=False)
                  for _ in range(2)
              ]
      else:
          models = [
              SimpleModel(hidden_dim, empty_grad=False)
              for _ in range(2)
          ]
  
      checkpoint_correctness_verification(
          config_dict,
          models,
          hidden_dim,
          tmpdir,
          load_optimizer_states=False,
      )
  
imports: import deepspeed
  from types import SimpleNamespace
  from deepspeed.ops.op_builder import CPUAdamBuilder
  from deepspeed.checkpoint.utils import clone_tensors_for_torch_save, get_model_ckpt_name_for_rank
  from deepspeed.accelerator import get_accelerator
  from deepspeed.runtime.utils import required_torch_version
  from unit.common import DistributedTest, DistributedFixture
  from unit.simple_model import *
  from unit.checkpoint.common import *
  import pytest
label_desc: Use image node_id 3 for calling the TestZeROCheckpoint obj's underlying member method code with example usage: obj.test_not_load_optimizer_state(tmpdir, zero_stage, use_cpu_offload, adam_optimizer) without return types