code: def test_raw(wrapper):
      raw = wrapper.raw()
      assert "model" in raw
      assert isinstance(raw["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 7 for calling a global function with example usage: test_raw(wrapper) without return types