code: def get_model_ckpt_name_for_rank(base_folder, mp_rank_str):
      ckpt_name = os.path.join(
          base_folder,
          MODEL_FILE_PREFIX + mp_rank_str + MODEL_FILE_SUFFIX,
      )
      return ckpt_name
  
imports: import os
  from .constants import MODEL_FILE_PREFIX, MODEL_FILE_SUFFIX, OPTIM_FILE_SUFFIX, ZERO_FILE_PREFIX
label_desc: Use image node_id 1 for calling a global function with example usage: get_model_ckpt_name_for_rank(base_folder, mp_rank_str) and returns: ckpt_name