code: def test_signature_for_pyfunc_predict_quantiles(
      auto_arima_model, model_path, data_airline, use_signature
  ):
      """Test saving of mlflow signature for pyfunc sktime predict_quantiles method."""
      model_path_primary = model_path.joinpath("primary")
      model_path_secondary = model_path.joinpath("secondary")
      flavor.save_model(
          sktime_model=auto_arima_model, path=model_path_primary
      )
      loaded_pyfunc = flavor.pyfunc.load_model(
          model_uri=model_path_primary
      )
      predict_conf = pd.DataFrame(
          [
              {
                  "fh": FH,
                  "predict_method": "predict_quantiles",
                  "alpha": ALPHA,
              }
          ]
      )
      forecast = loaded_pyfunc.predict(predict_conf)
      signature = (
          infer_signature(data_airline, forecast)
          if use_signature
          else None
      )
      flavor.save_model(
          auto_arima_model,
          path=model_path_secondary,
          signature=signature,
      )
      mlflow_model = Model.load(model_path_secondary)
      assert signature == mlflow_model.signature
  
imports: import os
  from pathlib import Path
  from unittest import mock
  import boto3
  import flavor
  import moto
  import numpy
  import pandas
  import pytest
  from botocore.config import Config
  from sktime.datasets import load_airline, load_longley
  from sktime.datatypes import convert
  from sktime.forecasting.arima import AutoARIMA
  from sktime.forecasting.model_selection import temporal_train_test_split
  from sktime.forecasting.naive import NaiveForecaster
  import mlflow
  from mlflow import pyfunc
  from mlflow.exceptions import MlflowException
  from mlflow.models import Model, infer_signature
  from mlflow.models.utils import _read_example
  from mlflow.store.artifact.s3_artifact_repo import S3ArtifactRepository
  from mlflow.tracking._model_registry import DEFAULT_AWAIT_MAX_SLEEP_SECONDS
  from mlflow.tracking.artifact_utils import _download_artifact_from_uri
  from mlflow.utils.environment import _mlflow_conda_env
label_desc: Use image node_id 14 for calling a global function with example usage: test_signature_for_pyfunc_predict_quantiles(auto_arima_model, model_path, data_airline, use_signature) without return types