class_docstr: Custom pyfunc class used to create customized mlflow models
code: def predict(
      self,
      context,
      model_input,
      params: Optional[Dict[str, Any]] = None,
  ):  # pylint: disable=unused-argument
      prediction = self.model.predict(model_input)
      return iris_classes(prediction)
  
file_path: mlflow/examples/pyfunc/train.py
imports: import os
  from typing import Any, Dict, Optional
  from custom_code import iris_classes
  from sklearn.datasets import load_iris
  from sklearn.linear_model import LogisticRegression
  import mlflow
  from mlflow.models import infer_signature
label_desc: Use image node_id 2 for calling the CustomPredict obj's underlying member method code with example usage: obj.predict(context, model_input, params) and returns: iris_classes