code: def test_checkpoint_fp32_optimizer(self, tmpdir):
      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": False},
      }
  
      args = args_from_dict(tmpdir, config_dict)
      hidden_dim = 10
      models = [
          SimpleModel(hidden_dim, empty_grad=False) for _ in range(2)
      ]
      checkpoint_correctness_verification(
          config_dict,
          models=models,
          hidden_dim=hidden_dim,
          tmpdir=tmpdir,
          fp16=False,
      )
  
file_path: DeepSpeed/tests/unit/checkpoint/test_other_optimizer.py
imports: import deepspeed
  from deepspeed.ops.op_builder import FusedLambBuilder
  from unit.common import DistributedTest
  from unit.simple_model import *
  from unit.checkpoint.common import checkpoint_correctness_verification
  import pytest
label_desc: Use image node_id 3 for calling the TestOtherOptimizerCheckpoint obj's underlying member method code with example usage: obj.test_checkpoint_fp32_optimizer(tmpdir) without return types