code: def __init__(self, throw_exception_on_batch_number=None) -> None:
      if throw_exception_on_batch_number is None:
          throw_exception_on_batch_number = []
      self.received_run_id = ""
      self.received_metrics = []
      self.received_tags = []
      self.received_params = []
      self.batch_count = 0
      self.throw_exception_on_batch_number = (
          throw_exception_on_batch_number
          if throw_exception_on_batch_number
          else []
      )
  
imports: import io
  import pickle
  import random
  import threading
  import time
  import uuid
  import pytest
  from mlflow import MlflowException
  from mlflow.entities.metric import Metric
  from mlflow.entities.param import Param
  from mlflow.entities.run_tag import RunTag
  from mlflow.utils.async_logging.async_logging_queue import AsyncLoggingQueue
label_desc: Use image node_id 1 to create a new RunData object  with example: obj = RunData(throw_exception_on_batch_number)