def __init__(
        self,
        data_folder: str,
        dataset_name: str,
        corpus_file: str = "corpus.jsonl",
        query_file: str = "queries.jsonl",
        qrels_folder: str = "qrels",
        qrels_file: str = "",
    ):
        """
        A utility class to load BEIR datasets into TruLens. Similar to https://github.com/beir-cellar/beir but with slightly
        more efficient implementation of processing and loading of the datasets using generators.

        Args:
            data_folder (str): local / remote path to store the downloaded dataset.
            dataset_name (str): Name of the dataset to be loaded. Must be one of https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/
            corpus_file (str, optional): file name of the corpus. Defaults to "corpus.jsonl".
            query_file (str, optional): file name of all the queries. Defaults to "queries.jsonl".
            qrels_folder (str, optional): folder name of qrels (relevance annotation). Defaults to "qrels".
            qrels_file (str, optional): file name of qrels (relevance annotation). Defaults to "".
        """
        self.qrels = {}

        self.dataset_name = dataset_name
        self.data_folder = data_folder

        self.corpus_file = (
            os.path.join(self.data_folder, dataset_name, corpus_file)
            if self.data_folder
            else corpus_file
        )
        self.query_file = (
            os.path.join(self.data_folder, dataset_name, query_file)
            if self.data_folder
            else query_file
        )
        self.qrels_folder = os.path.join(
            self.data_folder, dataset_name, qrels_folder
        )

        self.qrels_file = qrels_file

        A utility class to load BEIR datasets into TruLens. Similar to https://github.com/beir-cellar/beir but with slightly
        more efficient implementation of processing and loading of the datasets using generators.

        Args:
            data_folder (str): local / remote path to store the downloaded dataset.
            dataset_name (str): Name of the dataset to be loaded. Must be one of https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/
            corpus_file (str, optional): file name of the corpus. Defaults to "corpus.jsonl".
            query_file (str, optional): file name of all the queries. Defaults to "queries.jsonl".
            qrels_folder (str, optional): folder name of qrels (relevance annotation). Defaults to "qrels".
            qrels_file (str, optional): file name of qrels (relevance annotation). Defaults to "".
        
src/benchmark/trulens/benchmark/benchmark_frameworks/dataset/beir_loader.py