code: def test_load(wrapper):
      loaded_wrapper = TorchImageClassificationWrapper(
          "resnet18", {}, wrapper.out_dir
      )
      loaded_wrapper._load()
      assert isinstance(loaded_wrapper.model, nn.Module)
  
file_path: traintool/tests/image_classification/test_torch_models.py
imports: import comet_ml
  import pytest
  import numpy
  import torch.nn
  from torch import optim
  from tensorboardX import SummaryWriter
  from conftest import create_dataset, create_image
  from traintool.image_classification.torch_models import TorchImageClassificationWrapper, SimpleCnn
  from traintool.utils import DummyExperiment
label_desc: Use image node_id 5 for calling a global function with example usage: test_load(wrapper) without return types