A dataset wrapper around the evaluator inputs, designed for OpenCompass's internal use.
| 4 | |
| 5 | |
| 6 | class LMEvalDataset(BaseDataset): |
| 7 | """A dataset wrapper around the evaluator inputs, designed for |
| 8 | OpenCompass's internal use.""" |
| 9 | |
| 10 | @staticmethod |
| 11 | def load(**kwargs): |
| 12 | content = {k: v for k, v in kwargs.items() if v} |
| 13 | return DatasetDict(dict(test=Dataset.from_dict(content))) |