code: def set_predict(kwargs):
      """Set attr argument and proba"""
      out = dict()
      proba = kwargs.pop("proba", False)
      if proba:
          out["proba"] = proba
          out["attr"] = "predict_proba" if proba else "predict"
      return out
  
file_path: mlens/mlens/parallel/wrapper.py
imports: from ..None import config
  from .base import BaseParallel, OutputMixin
  from .backend import ParallelProcessing
  from ..utils.exceptions import ParallelProcessingError, NotFittedError
label_desc: Use image node_id 4 for calling a global function with example usage: set_predict(kwargs) and returns: out