code: def __init__(
      self,
      kind: Kind,
      e_id: int,
      t_id: int,
      s_id: int,
      num_batches: int,
      total_batches_processed: int,
  ) -> None:
      self.kind = kind
      self.experiment_id = e_id
      self.trial_id = t_id
      self.step_id = s_id
      self.num_batches = num_batches
      self.total_batches_processed = total_batches_processed
  
imports: import abc
  from enum import Enum, unique
  from typing import Any, Callable, Dict, Iterator, Optional, Tuple, Union, cast
  from determined.common import check
label_desc: Use image node_id 1 to create a new Workload object  with example: obj = Workload(kind, e_id, t_id, s_id, num_batches, total_batches_processed)