code: def __init__(
      self, ctx, max_length=1024, prompt_length=32, tokenizer=None
  ):
      self.ctx = ctx
      self.max_length = max_length + prompt_length
      self.prompt_length = prompt_length
      self.tokenizer = tokenizer
  
file_path: DeepKE/example/llm/CPM-Bee/src/cpm_live/training_tasks/ant/pretrain.py
imports: import torch.utils.data
  import random
  import numpy
label_desc: Use image node_id 1 to create a new CPMAntPretrainDataset object from inherited base classes: data with example: obj = CPMAntPretrainDataset(ctx, max_length, prompt_length, tokenizer)