utils partially refactored and tested

tasks/race/data.py

@@ -6,7 +6,7 @@ import time
 
 from torch.utils.data import Dataset
 
-from megatron.utils import print_rank_0
+from megatron import print_rank_0
 from tasks.data_utils import build_sample
 from tasks.data_utils import build_tokens_types_paddings_from_ids
 from tasks.data_utils import clean_text

tasks/race/finetune.py

 """Race."""
 
 from megatron.model.multiple_choice import MultipleChoice
-from megatron.utils import print_rank_0
+from megatron import print_rank_0
 from tasks.eval_utils import accuracy_func_provider
 from tasks.finetune_utils import finetune
 from tasks.race.data import RaceDataset