code: def _dist_search_fn_wrapper(args):
      return _dist_search_fn(*args)
  
imports: import contextlib
  import io
  import multiprocessing
  import os
  import re
  import time
  import keras_tuner
  import tensorflow
  from tensorflow import keras
  from tensorflow_cloud import CloudTuner
  from tensorflow_cloud.tuner import vizier_client
label_desc: Use image node_id 4 for calling a global function with example usage: _dist_search_fn_wrapper(args) and returns: _dist_search_fn