code: def make_data_parallel_mpu(
      dist_context: core.DistributedContext,
  ) -> ModelParallelUnit:
      return ModelParallelUnit(
          data_parallel_rank=dist_context.get_rank(),
          data_parallel_world_size=dist_context.get_size(),
          should_report_metrics=True,
          should_build_data_loader=True,
      )
  
imports: from dataclasses import dataclass
  from deepspeed.runtime.pipe import topology
  from determined import core
label_desc: Use image node_id 1 for calling a global function with example usage: make_data_parallel_mpu(dist_context) and returns: ModelParallelUnit