def __init__()

in scripts/adapet/ADAPET/src/data/DatasetReader.py [0:0]

• 28 lines of code
• 10 McCabe index (conditional complexity)

    def __init__(self, config, tokenizer, dataset):
        '''
        :param config:
        :param tokenizer:
        :param dataset:
        '''
        self.config = config
        self.dataset = dataset

        if self.dataset.lower() == "fewglue/boolq":
            self.dataset_reader = BoolQReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/cb":
            self.dataset_reader = CBReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/rte":
            self.dataset_reader = RTEReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/multirc":
            self.dataset_reader = MultiRCReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/wic":
            self.dataset_reader = WiCReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/copa":
            self.dataset_reader = COPAReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/record":
            self.dataset_reader = RecordReader(self.config, tokenizer)
        elif self.dataset.lower() == "fewglue/wsc":
            self.dataset_reader = WSCReader(self.config, tokenizer)
        elif self.dataset.lower() == "generic":
            self.dataset_reader = GenericReader(self.config, tokenizer)
        else:
            raise ValueError("Invalid Dataset name")