src/benchmarks/qa_datasets/__init__.py

from .amazon import AmazonSTaRKDataset
from .primekg import PrimeKGSTaRKDataset
from .mag import MAGSTaRKDataset
from .stark_qa import STaRKDataset