code: def export_setup_func_module(
      runtime_env: Union[Dict[str, Any], RuntimeEnv],
      setup_func_module: str,
  ) -> Union[Dict[str, Any], RuntimeEnv]:
      assert isinstance(setup_func_module, str)
      env_vars = runtime_env.get("env_vars", {})
      assert (
          ray_constants.WORKER_PROCESS_SETUP_HOOK_ENV_VAR
          not in env_vars
      ), (
          f"The env var, {ray_constants.WORKER_PROCESS_SETUP_HOOK_ENV_VAR}, "
          "is not permitted because it is reserved for the internal use."
      )
      env_vars[
          ray_constants.WORKER_PROCESS_SETUP_HOOK_ENV_VAR
      ] = setup_func_module
      runtime_env["env_vars"] = env_vars
      return runtime_env
  
file_path: ray/python/ray/_private/runtime_env/setup_hook.py
imports: import traceback
  import logging
  import base64
  import os
  from typing import Dict, Any, Callable, Union, Optional
  import ray
  import ray._private.ray_constants
  from ray._private.storage import _load_class
  import ray.cloudpickle
  from ray.runtime_env import RuntimeEnv
label_desc: Use image node_id 5 for calling a global function with example usage: export_setup_func_module(runtime_env, setup_func_module) and returns: runtime_env