text: from transformers import DistilBertTokenizer
  
  from .datamodule_base import BaseDataModule
  from ..datasets import NextQA
  
  
  class NextQADataModule(BaseDataModule):
      def __init__(self, *args, **kwargs):
          super().__init__(*args, **kwargs)
          self.ans_tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
  
      @property
      def dataset_cls(self):
          return NextQA
  
      @property
      def dataset_name(self):
          return "nqa"
  
      def setup(self, stage):
          super().setup(stage)
          self.train_dataset.ans_tokenizer = self.ans_tokenizer
          self.val_dataset.ans_tokenizer = self.ans_tokenizer
          self.test_dataset.ans_tokenizer = self.ans_tokenizer
  