class_docstr: The environment of the distributed training.
  
  Args:
      world_size: The number of total distributed training processes
      global_rank: The rank of the current process within this pool of training processes
class_docstr_tok: ["The","environment","of","the","distributed","training",".","Args",":","world_size",":","The","number","of","total","distributed","training","processes","global_rank",":","The","rank","of","the","current","process","within","this","pool","of","training","processes"]
code: def __init__(self, world_size: int, global_rank: int):
      self.world_size = world_size
      self.global_rank = global_rank
  
file_path: pytorch-lightning/src/lightning/data/utilities/env.py
imports: from typing import Callable, Optional
  import torch
  from torch.utils.data import get_worker_info
label_desc: Use image node_id 1 to create a new _DistributedEnv object  with example: obj = _DistributedEnv(world_size, global_rank)