src/benchmarks/qa_datasets/amazon.py

import os.path as osp
import json
import torch
from src.benchmarks.qa_datasets.stark_qa import STaRKDataset


class AmazonSTaRKDataset(STaRKDataset):
    
    def __init__(self, stark_qa_dir, split_dir):
        super().__init__(stark_qa_dir, split_dir)