code: def setup(task, model_class, train_raw, embedding):
      clear_session()  # prevent OOM during CI tests
      return mz.auto.prepare(
          task=task,
          model_class=model_class,
          data_pack=train_raw,
          embedding=embedding,
      )
  
file_path: MatchZoo/tests/unit_test/models/test_models.py
imports: import pytest
  import copy
  from pathlib import Path
  import shutil
  import matchzoo
  from keras.backend import clear_session
label_desc: Use image node_id 5 for calling a global function with example usage: setup(task, model_class, train_raw, embedding) and returns: mz