code: def test_models(backbone):
      num_classes = 13
      model = PointCloudSegmentation(
          backbone=backbone, num_classes=num_classes
      )
      assert model.head.weight.shape == torch.Size([13, 32])
  
file_path: lightning-flash/tests/pointcloud/segmentation/test_model.py
imports: import pytest
  import torch
  from flash.core.utilities.imports import _TOPIC_POINTCLOUD_AVAILABLE
  from flash.pointcloud.segmentation import PointCloudSegmentation
  from tests.helpers.task_tester import TaskTester
label_desc: Use image node_id 2 for calling a global function with example usage: test_models(backbone) without return types