Source code for lightning_ir.data.external_datasets.lsr_benchmark


[docs]
 1def register_lsr_dataset_to_ir_datasets(dataset):
 2    try:
 3        from lsr_benchmark import register_to_ir_datasets
 4    except ImportError:
 5        msg = (
 6            "I could not import the lsr_benchmark package. Please install the lsr-benchmark via "
 7            + f"'pip3 install lsr-benchmark' to load ir_datasets from there. I got the dataset id '{dataset}'."
 8        )
 9        print(msg)
10        raise ValueError(msg)
11
12    register_to_ir_datasets(dataset.replace("lsr-benchmark/", ""))