code: def count_rows(rows_per_file, path, day):
      day_file = os.path.join(path, f"day_{day}_labels.npy")
      data = np.load(day_file)
      num_rows = data.shape[0]
  
      rows_per_file[day] = num_rows
      print(f"counted {num_rows} for {day_file}")
  
file_path: torchrec/torchrec/datasets/scripts/shuffle_preproc_criteo.py
imports: import argparse
  import os
  import sys
  from multiprocessing import Manager, Process
  from typing import List
  import numpy
  from torchrec.datasets.criteo import BinaryCriteoUtils
label_desc: Use image node_id 2 for calling a global function with example usage: count_rows(rows_per_file, path, day) without return types