code: def multi_data_model(sagemaker_session):
      return MultiDataModel(
          name=MODEL_NAME,
          model_data_prefix=VALID_MULTI_MODEL_DATA_PREFIX,
          image_uri=IMAGE,
          role=ROLE,
          sagemaker_session=sagemaker_session,
      )
  
imports: from __future__ import absolute_import
  import os
  import pytest
  from mock import MagicMock, Mock, call, patch
  from sagemaker.multidatamodel import MULTI_MODEL_CONTAINER_MODE
  from sagemaker.multidatamodel import MultiDataModel
  from sagemaker.mxnet import MXNetModel, MXNetPredictor
label_desc: Use image node_id 2 for calling a global function with example usage: multi_data_model(sagemaker_session) and returns: MultiDataModel