code: def train_valid_datasets_provider():
      """Provide train and validation datasets."""
      args = get_args()
      tokenizer = get_tokenizer()
  
      train_dataset = RaceDataset(
          "training", args.train_data, tokenizer, args.seq_length
      )
      valid_dataset = RaceDataset(
          "validation", args.valid_data, tokenizer, args.seq_length
      )
  
      return train_dataset, valid_dataset
  
file_path: megatron-lm/tasks/race/finetune.py
imports: from megatron import get_args
  from megatron import print_rank_0
  from megatron import get_tokenizer
  from megatron.model.multiple_choice import MultipleChoice
  from tasks.eval_utils import accuracy_func_provider
  from tasks.finetune_utils import finetune
  from tasks.race.data import RaceDataset
  from megatron.arguments import core_transformer_config_from_args
label_desc: Use image node_id 1 for calling a global function with example usage: train_valid_datasets_provider() and returns: train_dataset,  valid_dataset