code: def create_deepspeed_args():
      parser = argparse.ArgumentParser()
      args = parser.parse_args(args="")
      args.deepspeed = True
      if dist.is_initialized():
          # We assume up to one full node executing unit tests
          assert dist.get_world_size() <= torch.cuda.device_count()
          args.local_rank = dist.get_rank()
      return args
  
imports: import os
  import json
  import argparse
  import torch
  from deepspeed.pipe import PipelineModule, LayerSpec
  from deepspeed.moe.layer import MoE
  import deepspeed.comm
label_desc: Use image node_id 5 for calling a global function with example usage: create_deepspeed_args() and returns: args