code: def _get_samples_per_class_semantic_segmentation(
      labels: List[np.ndarray],
  ) -> List[List[int]]:
      """Return a list containing the classes in batch."""
      return [np.unique(label).tolist() for label in labels]
  
file_path: deepchecks/deepchecks/vision/utils/label_prediction_properties.py
imports: from typing import List, Sequence, Union
  import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: _get_samples_per_class_semantic_segmentation(labels) and returns: unknown