code: def init_and_run(run_fn, run_config):
      os.environ["RANK"] = os.environ["SLURM_LOCALID"]
      os.environ["LOCAL_RANK"] = os.environ["SLURM_LOCALID"]
      os.environ["NODE_RANK"] = os.environ["SLURM_LOCALID"]
      os.environ["WORLD_SIZE"] = os.environ["SLURM_NTASKS"]
      run_fn(run_config)
  
imports: import os
  import pathlib
  import shutil
  import submitit
label_desc: Use image node_id 1 for calling a global function with example usage: init_and_run(run_fn, run_config) without return types