class_docstr: A wrapper class that should be used for testing purposes only.
class_docstr_tok: ["A","wrapper","class","that","should","be","used","for","testing","purposes","only","."]
docstr: Args:
      data: Model input data.
      params: Additional parameters to pass to the model for inference.
  
          .. Note:: Experimental: This parameter may change or be removed in a future
                     release without warning.
  
  Returns:
      Model predictions.
code: def predict(
      self,
      data,
      params: Optional[
          Dict[str, Any]
      ] = None,  # pylint: disable=unused-argument
  ):
      """
      Args:
          data: Model input data.
          params: Additional parameters to pass to the model for inference.
  
              .. Note:: Experimental: This parameter may change or be removed in a future
                         release without warning.
  
      Returns:
          Model predictions.
      """
      from mlflow.utils.openai_utils import _mock_openai_request
  
      with _mock_openai_request():
          return super().predict(data)
  
imports: import itertools
  import logging
  import os
  import warnings
  from string import Formatter
  from typing import Any, Dict, Optional, Set
  import yaml
  import mlflow
  from mlflow import pyfunc
  from mlflow.environment_variables import _MLFLOW_TESTING, MLFLOW_OPENAI_SECRET_SCOPE
  from mlflow.exceptions import MlflowException
  from mlflow.models import Model, ModelInputExample, ModelSignature
  from mlflow.models.model import MLMODEL_FILE_NAME
  from mlflow.models.utils import _save_example
  from mlflow.protos.databricks_pb2 import INVALID_PARAMETER_VALUE
  from mlflow.tracking._model_registry import DEFAULT_AWAIT_MAX_SLEEP_SECONDS
  from mlflow.tracking.artifact_utils import _download_artifact_from_uri
  from mlflow.types import ColSpec, Schema, TensorSpec
  from mlflow.utils.annotations import experimental
  from mlflow.utils.databricks_utils import check_databricks_secret_scope_access, is_in_databricks_runtime
  from mlflow.utils.docstring_utils import LOG_MODEL_PARAM_DOCS, format_docstring
  from mlflow.utils.environment import _CONDA_ENV_FILE_NAME, _CONSTRAINTS_FILE_NAME, _PYTHON_ENV_FILE_NAME, _REQUIREMENTS_FILE_NAME, _mlflow_conda_env, _process_conda_env, _process_pip_requirements, _PythonEnv, _validate_env_arguments
  from mlflow.utils.file_utils import write_to
  from mlflow.utils.model_utils import _add_code_from_conf_to_system_path, _get_flavor_configuration, _validate_and_copy_code_paths, _validate_and_prepare_target_save_path
  from mlflow.utils.openai_utils import REQUEST_URL_CHAT, REQUEST_URL_COMPLETIONS, REQUEST_URL_EMBEDDINGS, _OAITokenHolder, _OpenAIApiConfig, _OpenAIEnvVar, _validate_model_params
  from mlflow.utils.requirements_utils import _get_pinned_requirement
label_desc: Use image node_id 1 for calling the _TestOpenAIWrapper obj's underlying member method code with example usage: obj.predict(data, params) and returns: super