code: def override_env(env):
      old = os.environ
      try:
          os.environ = env
          yield
      finally:
          os.environ = old
  
imports: import contextlib
  import multiprocessing
  import os
  import shutil
  import sys
  import tempfile
  import time
  import traceback
  import cloudpickle
  import mock
  from horovod.runner.util.threads import in_thread
label_desc: Use image node_id 8 for calling a global function with example usage: override_env(env) without return types