code: def test_auto_arima_model_save_and_load(
      auto_arima_model, model_path, serialization_format
  ):
      """Test saving and loading of native sktime auto_arima_model."""
      flavor.save_model(
          sktime_model=auto_arima_model,
          path=model_path,
          serialization_format=serialization_format,
      )
      loaded_model = flavor.load_model(
          model_uri=model_path,
      )
  
      np.testing.assert_array_equal(
          auto_arima_model.predict(fh=FH), loaded_model.predict(fh=FH)
      )
  
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 8 for calling a global function with example usage: test_auto_arima_model_save_and_load(auto_arima_model, model_path, serialization_format) without return types