if (self.config['dataset'] is None) or (self.config['dataset'] not in ["wikitext-2", "wikitext-103"]):
            raise ConfigurationError("Task supports two 'dataset' options: 'wikitext-2', 'wikitext-103' ")