code: def _tok_function(
      self, raw_dataset, args: TrainArgs, format: str
  ) -> Dataset:
      raise NotImplementedError()
  
imports: import logging
  import os
  import tempfile
  from typing import Union
  from datasets import Dataset, DatasetDict, load_dataset, load_from_disk
  import torch
  from transformers import AutoConfig, AutoModel, AutoTokenizer, Seq2SeqTrainer, Seq2SeqTrainingArguments, Trainer, TrainingArguments
  from happytransformer.args import EvalArgs, TrainArgs
  from happytransformer.fine_tuning_util import EvalResult, FistStep, ZERO_2_SETTINGS, ZERO_3_SETTINGS
label_desc: Use image node_id 2 for calling the HappyTransformer obj's underlying member method code with example usage: obj._tok_function(raw_dataset, args, format) without return types