code: def wrap_softmax(softmax_fn):
      """Wraps softmax such that it outputs predictions and confidences
  
      Parameters
      ----------
      softmax_fn : fn
          Takes lists of inputs, outputs softmax probabilities (2d np.array)
  
      Returns
      -------
      function
          wrapped prediction function, returns (preds, confs) instead of softmax
  
      """
  
      def pred_and_conf(inputs):
          confs = softmax_fn(inputs)
          preds = np.argmax(confs, axis=1)
          return preds, confs
  
      pred_and_conf.conf = "softmax"
      return pred_and_conf
  
label_desc: Use image node_id 1 for calling the PredictorWrapper obj's underlying member method code with example usage: obj.wrap_softmax(softmax_fn) and returns: pred_and_conf, preds,  confs