code: def get_dataset(name):
      dataset_module = getattr(sm.datasets, name)
      data = dataset_module.load()
      data.exog = np.asarray(data.exog)
      data.endog = np.asarray(data.endog)
      return data
  
imports: from collections import namedtuple
  import numpy
  import pandas
  import statsmodels.api
  from scipy.linalg import toeplitz
  from statsmodels.tsa.arima.model import ARIMA
  from statsmodels.tsa.arima_process import arma_generate_sample
  from mlflow.models import ModelSignature
  from mlflow.types.schema import Schema, TensorSpec
label_desc: Use image node_id 4 for calling a global function with example usage: get_dataset(name) and returns: data