code: def get_world_size() -> int:
      """
      Simple wrapper for correctly getting worldsize in both distributed
      / non-distributed settings
      """
      return (
          torch.distributed.get_world_size()
          if torch.distributed.is_available()
          and torch.distributed.is_initialized()
          else 1
      )
  
imports: import io
  import tempfile
  from typing import Any, Callable, List, Tuple
  import torch
label_desc: Use image node_id 14 for calling a global function with example usage: get_world_size() and returns: unknown