code: def _get_predicted_classes_per_image_classification(
      predictions: List[np.ndarray],
  ) -> List[int]:
      """Return a list of the predicted class per image in the batch."""
      return np.argmax(predictions, axis=1).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 8 for calling a global function with example usage: _get_predicted_classes_per_image_classification(predictions) and returns: np