code: def test_initialization_of_vision_data_with_classification_dataset_that_contains_incorrect_labels():
      # Arrange
      loader_with_string_labels = DataLoader(
          dataset=[
              (torch.ones((3, 3, 3)) * 2, "a"),
              (torch.ones((3, 3, 3)) * 2, "b"),
          ],
          collate_fn=_simple_batch_collate,
      )
      loader_with_labels_of_incorrect_shape = DataLoader(
          dataset=[
              (torch.ones((3, 3, 3)) * 2, torch.tensor([1, 2])),
              (torch.ones((3, 3, 3)) * 2, torch.tensor([2, 3])),
          ],
          collate_fn=_simple_batch_collate,
      )
  
      # Assert
      assert_that(
          calling(VisionData).with_args(
              loader_with_string_labels, TaskType.CLASSIFICATION.value
          ),
          raises(
              ValidationError,
              r"Classification label per image must be a number.",
          ),
      )
      assert_that(
          calling(VisionData).with_args(
              loader_with_labels_of_incorrect_shape,
              TaskType.CLASSIFICATION.value,
          ),
          raises(
              ValidationError,
              r"Classification label per image must be a number.",
          ),
      )
  
imports: import numpy
  import pytest
  import torch
  from hamcrest import assert_that, calling, equal_to, has_length, is_not, raises
  from torch.utils.data import DataLoader
  from deepchecks.core.errors import DatasetValidationError, ValidationError
  from deepchecks.vision.datasets.classification.mnist_torch import IterableTorchMnistDataset, collate_without_model
  from deepchecks.vision.datasets.detection import coco_torch
  from deepchecks.vision.datasets.segmentation import segmentation_coco
  from deepchecks.vision.utils.test_utils import replace_collate_fn_dataloader
  from deepchecks.vision.vision_data import TaskType, VisionData
  from deepchecks.vision.vision_data.utils import validate_vision_data_compatibility
  from tests.vision.conftest import run_update_loop
label_desc: Use image node_id 6 for calling a global function with example usage: test_initialization_of_vision_data_with_classification_dataset_that_contains_incorrect_labels() without return types