code: def from_json(data: Dict[str, Any]) -> "Workload":
      check.check_in(data["kind"], Workload.Kind.__members__)
      return Workload(
          Workload.Kind[data["kind"]],
          data["experiment_id"],
          data["trial_id"],
          data["step_id"],
          data["num_batches"],
          data["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 6 for calling the Workload obj's underlying member method code with example usage: obj.from_json(data) and returns: Workload