code: def small_wikiann_train_test_text_data(original_wikiann):
      """Wikiann dataset for token classification"""
      train = original_wikiann["train"][:50]
      test = original_wikiann["test"][:50]
      return _wikiann_to_text_data(train), _wikiann_to_text_data(test)
  
imports: import random
  import typing
  import numpy
  import pytest
  from datasets import load_dataset
  from nltk import download
  from nltk.corpus import movie_reviews
  from deepchecks.nlp.datasets.classification import just_dance_comment_analysis, tweet_emotion
  from deepchecks.nlp.text_data import TextData
label_desc: Use image node_id 21 for calling a global function with example usage: small_wikiann_train_test_text_data(original_wikiann) and returns: _wikiann_to_text_data,  _wikiann_to_text_data