code: def print_processing_stats(
      self, count, proc_start, total_bytes_processed
  ):
      if count % self.args.log_interval == 0:
          current = time.time()
          elapsed = current - proc_start
          mbs = total_bytes_processed / elapsed / 1024 / 1024
          print(
              f"Processed {count} documents",
              f"({count/elapsed} docs/s, {mbs} MB/s).",
              file=sys.stderr,
          )
  
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 the Partition obj's underlying member method code with example usage: obj.print_processing_stats(count, proc_start, total_bytes_processed) without return types