content: from .base_dataset import BaseDataset
  
  
  class F30KCaptionKarpathyDataset(BaseDataset):
      def __init__(self, *args, split="", **kwargs):
          assert split in ["train", "val", "test"]
  
          if split == "train":
              names = ["f30k_caption_karpathy_train", "f30k_caption_karpathy_val"]
          elif split == "val":
              names = ["f30k_caption_karpathy_test"]
          elif split == "test":
              names = ["f30k_caption_karpathy_test"]
  
          super().__init__(*args, **kwargs, names=names, text_column_name="caption")
  
      def __getitem__(self, index):
          return self.get_suite(index)
  