code: def get_file_name(args, file_id):
      file_name, extension = os.path.splitext(args.input)
      input_file_name = file_name + "_" + str(file_id) + extension
      sentence_split_file = (
          file_name + "_ss_" + str(file_id) + extension
      )
      output_prefix = args.output_prefix + "_" + str(file_id)
      file_names = {
          "partition": input_file_name,
          "sentence_split": sentence_split_file,
          "output_prefix": output_prefix,
      }
      return file_names
  
imports: import argparse
  import math
  import json
  import os
  import sys
  import time
  import gzip
  import glob
  import torch
  import numpy
  import multiprocessing
  from megatron.tokenizer import build_tokenizer
  from megatron.core.datasets import indexed_dataset
label_desc: Use image node_id 2 for calling a global function with example usage: get_file_name(args, file_id) and returns: file_names