code: def _get_samples_per_class_classification(
      labels: Union[np.ndarray, List]
  ) -> List[int]:
      """Return a list containing the class per image in batch."""
      return labels if isinstance(labels, List) else labels.tolist()
  
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 1 for calling a global function with example usage: _get_samples_per_class_classification(labels) and returns: unknown