code: def compile_args(config):
      if "lr" in config:
          lr = config["lr"]
      else:
          lr = 1e-3
      args = {
          "optimizer": tf.keras.optimizers.SGD(lr),
          "loss": "mean_squared_error",
          "metrics": ["mean_squared_error"],
      }
      return args
  
file_path: BigDL/python/orca/test/bigdl/orca/learn/ray/ctx/test_tf2_ray_runtime.py
imports: import pytest
  from unittest import TestCase
  import numpy
  import tensorflow
  from bigdl.orca import init_orca_context, stop_orca_context
  from bigdl.orca.learn.tf2 import Estimator
label_desc: Use image node_id 5 for calling a global function with example usage: compile_args(config) and returns: args