code: def get_dataset(cfg, name):
      if cfg.data.category == "cars" or cfg.data.category == "chairs":
          return SRNDataset(cfg, name)
      elif (
          cfg.data.category == "hydrants"
          or cfg.data.category == "teddybears"
      ):
          return CO3DDataset(cfg, name)
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_dataset(cfg, name) and returns: SRNDataset, CO3DDataset