code: def test_saved_model_support_array_type_input():
      def infer(features):
          res = np.expand_dims(features.numpy().sum(axis=1), axis=1)
          return {"prediction": tf.constant(res)}
  
      model = _TF2Wrapper(None, infer)
      infer_df = pd.DataFrame(
          {"features": [[1.0, 2.0, 3.0, 4.0], [5.0, 6.0, 7.0, 8.0]]}
      )
  
      result = model.predict(infer_df)
  
      np.testing.assert_allclose(
          result["prediction"], infer_df.applymap(sum).values[:, 0]
      )
  
imports: import collections
  import json
  import os
  import iris_data_utils
  import numpy
  import pandas
  import pandas.testing
  import pytest
  import tensorflow
  import mlflow
  import mlflow.pyfunc.scoring_server
  from mlflow import pyfunc
  from mlflow.store.artifact.s3_artifact_repo import S3ArtifactRepository
  from mlflow.tensorflow import _TF2Wrapper
  from mlflow.utils.conda import get_or_create_conda_env
  from mlflow.utils.environment import _mlflow_conda_env
  from tests.helper_functions import assert_array_almost_equal, pyfunc_serve_and_score_model
label_desc: Use image node_id 10 for calling a global function with example usage: test_saved_model_support_array_type_input() and returns: dict