code: def _get_data_parallel_world_size():
      """Return world size for the data parallel group."""
      global mpu
      if mpu is not None:
          return mpu.get_data_parallel_world_size()
      return dist.get_world_size(group=_get_data_parallel_group())
  
imports: from deepspeed import comm
  from deepspeed.utils import log_dist
  from deepspeed.utils.exceptions import DeprecatedException
label_desc: Use image node_id 23 for calling a global function with example usage: _get_data_parallel_world_size() and returns: dist, mpu