code: def test_yolo_model(backbone):
      model = models.ultralytics.yolov5.model(
          num_classes=5,
          img_size=320,
          backbone=backbone(pretrained=True),
      )
      weights_path = (
          get_root_dir() / "yolo" / f"{backbone.model_name}.pt"
      )
  
      assert weights_path.is_file() == True
      assert len(list(model.param_groups())) == 3
      assert model.nc == 4
  
imports: import pytest
  from icevision.all import *
  from icevision.models.ultralytics.yolov5.backbones import *
label_desc: Use image node_id 1 for calling a global function with example usage: test_yolo_model(backbone) without return types