code: def test_zoo_load(self):
      with mock_env_with_temp():
          checkpoint = Checkpoint(self.trainer)
          self._init_early_stopping(checkpoint)
          self._do_a_pass()
  
          original_model = deepcopy(self.trainer.model)
          ret_load_pretrained_zoo = {
              "config": self.config.model_config,
              "checkpoint": deepcopy(self.trainer.model.state_dict()),
              "full_config": self.config,
          }
  
          self._do_a_pass()
  
          with patch(
              "mmf.utils.checkpoint.load_pretrained_model",
              return_value=ret_load_pretrained_zoo,
          ):
              self.trainer.config.checkpoint.resume_zoo = "random"
              with contextlib.redirect_stdout(StringIO()):
                  checkpoint.load_state_dict()
              self.assertTrue(
                  compare_state_dicts(
                      self.trainer.model.state_dict(),
                      original_model.state_dict(),
                  )
              )
  
              # Now, test zoo override
              self.trainer.config.checkpoint.zoo_config_override = True
              SimpleModule.from_pretrained = Mock(
                  return_value=deepcopy(original_model)
              )
              registry.register_model("simple")(SimpleModule)
              with contextlib.redirect_stdout(StringIO()):
                  checkpoint.load_state_dict()
              self.assertTrue(
                  compare_state_dicts(
                      self.trainer.model.state_dict(),
                      original_model.state_dict(),
                  )
              )
  
imports: import contextlib
  import os
  import tempfile
  import unittest
  from copy import deepcopy
  from io import StringIO
  from unittest.mock import Mock, patch
  import torch
  from mmf.common.registry import registry
  from mmf.models.base_model import BaseModel
  from mmf.trainers.callbacks.checkpoint import CheckpointCallback
  from mmf.trainers.callbacks.early_stopping import EarlyStoppingCallback
  from mmf.trainers.callbacks.lr_scheduler import LRSchedulerCallback
  from mmf.utils.checkpoint import Checkpoint
  from mmf.utils.configuration import load_yaml
  from mmf.utils.file_io import PathManager
  from omegaconf import OmegaConf
  from tests.test_utils import compare_state_dicts, skip_if_no_cuda
label_desc: Use image node_id 9 for calling the TestUtilsCheckpoint obj's underlying member method code with example usage: obj.test_zoo_load() without return types