code: def convert_tsv_to_parquet(input_path: str, output_base_path: str):
      config = {
          "engine": "csv",
          "names": DEFAULT_COLUMN_NAMES,
          "sep": "\t",
          "dtypes": dtypes,
          "part_size": "128MB",
      }
  
      output_path = os.path.join(output_base_path, "criteo_parquet")
      if os.path.exists(output_path):
          shutil.rmtree(output_path)
      os.makedirs(output_path)
  
      # split last day into two parts
      number_of_lines = int(
          subprocess.check_output(
              (f'wc -l {os.path.join(input_path, "day_23")}').split()
          ).split()[0]
      )
      valid_set_size = number_of_lines // 2
      test_set_size = number_of_lines - valid_set_size
  
      start_time = time.time()
      with open(os.path.join(input_path, "day_23.part0"), "w") as f:
          subprocess.run(
              [
                  "head",
                  "-n",
                  str(test_set_size),
                  str(os.path.join(input_path, "day_23")),
              ],
              stdout=f,
          )
  
      with open(os.path.join(input_path, "day_23.part1"), "w") as f:
          subprocess.run(
              [
                  "tail",
                  "-n",
                  str(valid_set_size),
                  str(os.path.join(input_path, "day_23")),
              ],
              stdout=f,
          )
  
      print(
          f"finished splitting the last day, took {time.time() - start_time}"
      )
  
      input_paths = [
          os.path.join(input_path, f"day_{day}")
          for day in range(DAYS - 1)
      ] + [
          os.path.join(input_path, f"day_23.part{i}") for i in range(2)
      ]
  
      print(f"handling the input paths: {input_paths}")
  
      tsv_dataset = nvt.Dataset(input_paths, **config)
  
      print("finished loading the tsv dataset")
  
      tsv_dataset.to_parquet(
          output_path,
          preserve_files=True,
      )
  
file_path: torchrec/torchrec/datasets/scripts/nvt/convert_tsv_to_parquet.py
imports: import argparse
  import os
  import shutil
  import subprocess
  import time
  import numpy
  import nvtabular
  from utils.criteo_constant import DAYS, DEFAULT_COLUMN_NAMES, DEFAULT_LABEL_NAME
  from utils.dask import setup_dask
label_desc: Use image node_id 1 for calling a global function with example usage: convert_tsv_to_parquet(input_path, output_base_path) without return types