code: def process_data(
      ds: DatasetDict, vocabulary: Vocabulary, max_len=256
  ) -> DatasetDict:
      # 分词
      ds = ds.map(
          lambda x: {
              "input_ids": text_to_id(vocabulary, x["text"], max_len)
          }
      )
      ds.set_format(type="numpy", columns=ds.column_names)
      return ds
  
imports: import os
  import sys
  import time
  from datasets import load_dataset
  from datasets import DatasetDict
  import jittor
  from jittor import nn, Module
  from jittor.dataset import Dataset
  from fastNLP.core.controllers.trainer import Trainer
  from fastNLP.core.metrics.accuracy import Accuracy
  from fastNLP.core.vocabulary import Vocabulary
  from fastNLP.core.callbacks.progress_callback import RichCallback
  from fastNLP.core.callbacks.callback import Callback
  from fastNLP.core.dataloaders.jittor_dataloader.fdl import JittorDataLoader
label_desc: Use image node_id 1 for calling a global function with example usage: process_data(ds, vocabulary, max_len) and returns: ds