code: def create_stop(stop_metric=None):
      stop = {"training_iteration": 20}
      stop.update({"reward_metric": stop_metric})
      return stop
  
imports: from test.zoo.pipeline.utils.test_utils import ZooTestCase
  from zoo.orca.automl.search import SearchEngineFactory
  from zoo.orca.automl.search.ray_tune import RayTuneSearchEngine
  from zoo.orca.automl.model.base_pytorch_model import PytorchModelBuilder
  import torch
  import torch.nn
  from zoo.orca.automl import hp
  import numpy
  from zoo.orca import init_orca_context, stop_orca_context
label_desc: Use image node_id 2 for calling a global function with example usage: create_stop(stop_metric) and returns: stop