code: def __init__(self, wandb_config):
      super().__init__(wandb_config)
      check_wandb_availability()
      import wandb
  
      self.enabled = wandb_config.enabled
      self.group = wandb_config.group
      self.team = wandb_config.team
      self.project = wandb_config.project
  
      if self.enabled and dist.get_rank() == 0:
          wandb.init(
              project=self.project, group=self.group, entity=self.team
          )
  
file_path: DeepSpeed/deepspeed/monitor/wandb.py
imports: from .utils import check_wandb_availability
  from .monitor import Monitor
  import deepspeed.comm
label_desc: Use image node_id 1 to create a new WandbMonitor object from inherited base classes: Monitor with example: obj = WandbMonitor(wandb_config)