├── .devcontainer ├── Dockerfile.dev └── devcontainer.json ├── .github ├── dependabot.yml └── workflows │ ├── builddocs.yml │ ├── linters.yml │ ├── python-publish.yml │ ├── test-models.yml │ └── tests.yml ├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── configs ├── callbacks │ ├── index-faiss-flat.yaml │ ├── index-faiss-ivf.yaml │ ├── index-faiss-ivfpq.yaml │ ├── index-sparse.yaml │ ├── re-rank.yaml │ └── regularization-scheduler.yaml ├── data │ ├── fine-tune-msmarco.yaml │ ├── index-msmarco.yaml │ ├── rank-distillm.yaml │ ├── re-rank-trec-dl.yaml │ ├── run-dataset.yaml │ └── search-trec-dl.yaml ├── examples │ ├── fine-tune.yaml │ ├── index.yaml │ ├── re-rank.yaml │ └── search.yaml ├── models │ ├── bi-encoder.yaml │ ├── colbert.yaml │ ├── cross-encoder.yaml │ └── monoelectra.yaml ├── optimizer.yaml └── trainer │ ├── finetune.yaml │ └── inference.yaml ├── docs ├── Makefile ├── _static │ ├── css │ │ └── custom.css │ └── lightning-ir-logo.svg ├── _templates │ └── autosummary │ │ ├── base.rst │ │ ├── class.rst │ │ └── module.rst ├── api.rst ├── concepts │ ├── data.rst │ ├── model.rst │ └── trainer.rst ├── conf.py ├── core.rst ├── howto.rst ├── howto │ ├── dataset.rst │ └── model.rst ├── index.rst ├── make.bat ├── model-zoo.rst ├── models │ ├── nanobeir_details_bienc.csv │ ├── nanobeir_details_colbert.csv │ ├── nanobeir_details_external_bienc.csv │ ├── nanobeir_details_external_colbert.csv │ ├── nanobeir_details_external_moderncol.csv │ ├── nanobeir_details_external_splade.csv │ ├── nanobeir_details_splade.csv │ ├── reranking.csv │ └── retrieval.csv └── quickstart.rst ├── examples ├── adapters.py ├── custom_bi_encoder.py ├── custom_cross_encoder.py ├── datamodule.py ├── doc_dataset.py ├── fine_tune.py ├── index.py ├── query_dataset.py ├── re_rank.py ├── run_dataset.py ├── scoring.py ├── search.py └── tuple_dataset.py ├── lightning_ir ├── __init__.py ├── __main__.py ├── base │ ├── __init__.py │ ├── adapter.py │ ├── class_factory.py │ ├── config.py │ ├── external_model_hub.py │ ├── model.py │ ├── module.py │ ├── tokenizer.py │ └── validation_utils.py ├── bi_encoder │ ├── __init__.py │ ├── bi_encoder_config.py │ ├── bi_encoder_model.py │ ├── bi_encoder_module.py │ └── bi_encoder_tokenizer.py ├── callbacks │ ├── __init__.py │ └── callbacks.py ├── cross_encoder │ ├── __init__.py │ ├── cross_encoder_config.py │ ├── cross_encoder_model.py │ ├── cross_encoder_module.py │ └── cross_encoder_tokenizer.py ├── data │ ├── __init__.py │ ├── data.py │ ├── datamodule.py │ ├── dataset.py │ └── external_datasets │ │ ├── __init__.py │ │ ├── colbert.py │ │ ├── cross_architecture_knowledge_distillation.py │ │ ├── ir_datasets_utils.py │ │ ├── lsr_benchmark.py │ │ ├── rank_distillm.py │ │ └── sbert.py ├── loss │ ├── __init__.py │ ├── approximate.py │ ├── base.py │ ├── embedding.py │ ├── in_batch.py │ ├── listwise.py │ ├── neural.py │ ├── pairwise.py │ └── regularization.py ├── main.py ├── modeling_utils │ ├── __init__.py │ ├── batching.py │ ├── embedding_post_processing.py │ └── lm_head.py ├── models │ ├── __init__.py │ ├── bi_encoders │ │ ├── __init__.py │ │ ├── coil.py │ │ ├── col.py │ │ ├── dpr.py │ │ ├── mvr.py │ │ └── splade.py │ ├── cross_encoders │ │ ├── __init__.py │ │ ├── mono.py │ │ └── set_encoder.py │ ├── register_external_models.py │ └── register_internal_models.py ├── retrieve │ ├── __init__.py │ ├── base │ │ ├── __init__.py │ │ ├── indexer.py │ │ ├── packed_tensor.py │ │ └── searcher.py │ ├── faiss │ │ ├── __init__.py │ │ ├── faiss_indexer.py │ │ └── faiss_searcher.py │ ├── plaid │ │ ├── __init__.py │ │ ├── plaid_indexer.py │ │ └── plaid_searcher.py │ ├── pytorch │ │ ├── __init__.py │ │ ├── dense_indexer.py │ │ ├── dense_searcher.py │ │ ├── sparse_indexer.py │ │ └── sparse_searcher.py │ └── seismic │ │ ├── __init__.py │ │ ├── seismic_format.py │ │ ├── seismic_indexer.py │ │ └── seismic_searcher.py └── schedulers │ ├── __init__.py │ ├── lr_schedulers.py │ └── schedulers.py ├── pyproject.toml └── tests ├── __init__.py ├── conftest.py ├── data ├── corpus │ ├── docpairs.tsv │ ├── docs.tsv │ ├── docs.tsv.pklz4 │ │ ├── bin │ │ ├── bin.meta │ │ ├── bin.pos │ │ ├── idx.doc_id.key │ │ ├── idx.doc_id.meta │ │ └── idx.doc_id.pos │ ├── qrels.tsv │ └── queries.tsv ├── runs │ ├── lightning-ir.tsv │ └── run.jsonl └── test_model │ ├── config.json │ ├── model.safetensors │ ├── special_tokens_map.json │ ├── tokenizer.json │ ├── tokenizer_config.json │ └── vocab.txt ├── test_callbacks.py ├── test_config.py ├── test_data.py ├── test_external_datasets.py ├── test_loss.py ├── test_lsr_benchmark_integration.py ├── test_model.py ├── test_models ├── __init__.py ├── test_coil.py ├── test_col.py ├── test_dpr.py ├── test_mono.py ├── test_mvr.py ├── test_set_encoder.py └── test_splade.py ├── test_schedulers.py └── test_tokenizer.py /.devcontainer/Dockerfile.dev: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.devcontainer/Dockerfile.dev -------------------------------------------------------------------------------- /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/builddocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.github/workflows/builddocs.yml -------------------------------------------------------------------------------- /.github/workflows/linters.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.github/workflows/linters.yml -------------------------------------------------------------------------------- /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.github/workflows/test-models.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.github/workflows/test-models.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/README.md -------------------------------------------------------------------------------- /configs/callbacks/index-faiss-flat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/callbacks/index-faiss-flat.yaml -------------------------------------------------------------------------------- /configs/callbacks/index-faiss-ivf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/callbacks/index-faiss-ivf.yaml -------------------------------------------------------------------------------- /configs/callbacks/index-faiss-ivfpq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/callbacks/index-faiss-ivfpq.yaml -------------------------------------------------------------------------------- /configs/callbacks/index-sparse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/callbacks/index-sparse.yaml -------------------------------------------------------------------------------- /configs/callbacks/re-rank.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/callbacks/re-rank.yaml -------------------------------------------------------------------------------- /configs/callbacks/regularization-scheduler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/callbacks/regularization-scheduler.yaml -------------------------------------------------------------------------------- /configs/data/fine-tune-msmarco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/data/fine-tune-msmarco.yaml -------------------------------------------------------------------------------- /configs/data/index-msmarco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/data/index-msmarco.yaml -------------------------------------------------------------------------------- /configs/data/rank-distillm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/data/rank-distillm.yaml -------------------------------------------------------------------------------- /configs/data/re-rank-trec-dl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/data/re-rank-trec-dl.yaml -------------------------------------------------------------------------------- /configs/data/run-dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/data/run-dataset.yaml -------------------------------------------------------------------------------- /configs/data/search-trec-dl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/data/search-trec-dl.yaml -------------------------------------------------------------------------------- /configs/examples/fine-tune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/examples/fine-tune.yaml -------------------------------------------------------------------------------- /configs/examples/index.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/examples/index.yaml -------------------------------------------------------------------------------- /configs/examples/re-rank.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/examples/re-rank.yaml -------------------------------------------------------------------------------- /configs/examples/search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/examples/search.yaml -------------------------------------------------------------------------------- /configs/models/bi-encoder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/models/bi-encoder.yaml -------------------------------------------------------------------------------- /configs/models/colbert.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/models/colbert.yaml -------------------------------------------------------------------------------- /configs/models/cross-encoder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/models/cross-encoder.yaml -------------------------------------------------------------------------------- /configs/models/monoelectra.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/models/monoelectra.yaml -------------------------------------------------------------------------------- /configs/optimizer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/optimizer.yaml -------------------------------------------------------------------------------- /configs/trainer/finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/trainer/finetune.yaml -------------------------------------------------------------------------------- /configs/trainer/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/configs/trainer/inference.yaml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/_static/lightning-ir-logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/_static/lightning-ir-logo.svg -------------------------------------------------------------------------------- /docs/_templates/autosummary/base.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/_templates/autosummary/base.rst -------------------------------------------------------------------------------- /docs/_templates/autosummary/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/_templates/autosummary/class.rst -------------------------------------------------------------------------------- /docs/_templates/autosummary/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/_templates/autosummary/module.rst -------------------------------------------------------------------------------- /docs/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/api.rst -------------------------------------------------------------------------------- /docs/concepts/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/concepts/data.rst -------------------------------------------------------------------------------- /docs/concepts/model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/concepts/model.rst -------------------------------------------------------------------------------- /docs/concepts/trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/concepts/trainer.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/core.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/core.rst -------------------------------------------------------------------------------- /docs/howto.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/howto.rst -------------------------------------------------------------------------------- /docs/howto/dataset.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/howto/dataset.rst -------------------------------------------------------------------------------- /docs/howto/model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/howto/model.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/model-zoo.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/model-zoo.rst -------------------------------------------------------------------------------- /docs/models/nanobeir_details_bienc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_bienc.csv -------------------------------------------------------------------------------- /docs/models/nanobeir_details_colbert.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_colbert.csv -------------------------------------------------------------------------------- /docs/models/nanobeir_details_external_bienc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_external_bienc.csv -------------------------------------------------------------------------------- /docs/models/nanobeir_details_external_colbert.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_external_colbert.csv -------------------------------------------------------------------------------- /docs/models/nanobeir_details_external_moderncol.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_external_moderncol.csv -------------------------------------------------------------------------------- /docs/models/nanobeir_details_external_splade.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_external_splade.csv -------------------------------------------------------------------------------- /docs/models/nanobeir_details_splade.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/nanobeir_details_splade.csv -------------------------------------------------------------------------------- /docs/models/reranking.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/reranking.csv -------------------------------------------------------------------------------- /docs/models/retrieval.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/models/retrieval.csv -------------------------------------------------------------------------------- /docs/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/docs/quickstart.rst -------------------------------------------------------------------------------- /examples/adapters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/adapters.py -------------------------------------------------------------------------------- /examples/custom_bi_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/custom_bi_encoder.py -------------------------------------------------------------------------------- /examples/custom_cross_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/custom_cross_encoder.py -------------------------------------------------------------------------------- /examples/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/datamodule.py -------------------------------------------------------------------------------- /examples/doc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/doc_dataset.py -------------------------------------------------------------------------------- /examples/fine_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/fine_tune.py -------------------------------------------------------------------------------- /examples/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/index.py -------------------------------------------------------------------------------- /examples/query_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/query_dataset.py -------------------------------------------------------------------------------- /examples/re_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/re_rank.py -------------------------------------------------------------------------------- /examples/run_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/run_dataset.py -------------------------------------------------------------------------------- /examples/scoring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/scoring.py -------------------------------------------------------------------------------- /examples/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/search.py -------------------------------------------------------------------------------- /examples/tuple_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/examples/tuple_dataset.py -------------------------------------------------------------------------------- /lightning_ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/__init__.py -------------------------------------------------------------------------------- /lightning_ir/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/__main__.py -------------------------------------------------------------------------------- /lightning_ir/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/__init__.py -------------------------------------------------------------------------------- /lightning_ir/base/adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/adapter.py -------------------------------------------------------------------------------- /lightning_ir/base/class_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/class_factory.py -------------------------------------------------------------------------------- /lightning_ir/base/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/config.py -------------------------------------------------------------------------------- /lightning_ir/base/external_model_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/external_model_hub.py -------------------------------------------------------------------------------- /lightning_ir/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/model.py -------------------------------------------------------------------------------- /lightning_ir/base/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/module.py -------------------------------------------------------------------------------- /lightning_ir/base/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/tokenizer.py -------------------------------------------------------------------------------- /lightning_ir/base/validation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/base/validation_utils.py -------------------------------------------------------------------------------- /lightning_ir/bi_encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/bi_encoder/__init__.py -------------------------------------------------------------------------------- /lightning_ir/bi_encoder/bi_encoder_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/bi_encoder/bi_encoder_config.py -------------------------------------------------------------------------------- /lightning_ir/bi_encoder/bi_encoder_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/bi_encoder/bi_encoder_model.py -------------------------------------------------------------------------------- /lightning_ir/bi_encoder/bi_encoder_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/bi_encoder/bi_encoder_module.py -------------------------------------------------------------------------------- /lightning_ir/bi_encoder/bi_encoder_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/bi_encoder/bi_encoder_tokenizer.py -------------------------------------------------------------------------------- /lightning_ir/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/callbacks/__init__.py -------------------------------------------------------------------------------- /lightning_ir/callbacks/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/callbacks/callbacks.py -------------------------------------------------------------------------------- /lightning_ir/cross_encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/cross_encoder/__init__.py -------------------------------------------------------------------------------- /lightning_ir/cross_encoder/cross_encoder_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/cross_encoder/cross_encoder_config.py -------------------------------------------------------------------------------- /lightning_ir/cross_encoder/cross_encoder_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/cross_encoder/cross_encoder_model.py -------------------------------------------------------------------------------- /lightning_ir/cross_encoder/cross_encoder_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/cross_encoder/cross_encoder_module.py -------------------------------------------------------------------------------- /lightning_ir/cross_encoder/cross_encoder_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/cross_encoder/cross_encoder_tokenizer.py -------------------------------------------------------------------------------- /lightning_ir/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/__init__.py -------------------------------------------------------------------------------- /lightning_ir/data/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/data.py -------------------------------------------------------------------------------- /lightning_ir/data/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/datamodule.py -------------------------------------------------------------------------------- /lightning_ir/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/dataset.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/__init__.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/colbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/colbert.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/cross_architecture_knowledge_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/cross_architecture_knowledge_distillation.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/ir_datasets_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/ir_datasets_utils.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/lsr_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/lsr_benchmark.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/rank_distillm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/rank_distillm.py -------------------------------------------------------------------------------- /lightning_ir/data/external_datasets/sbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/data/external_datasets/sbert.py -------------------------------------------------------------------------------- /lightning_ir/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/__init__.py -------------------------------------------------------------------------------- /lightning_ir/loss/approximate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/approximate.py -------------------------------------------------------------------------------- /lightning_ir/loss/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/base.py -------------------------------------------------------------------------------- /lightning_ir/loss/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/embedding.py -------------------------------------------------------------------------------- /lightning_ir/loss/in_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/in_batch.py -------------------------------------------------------------------------------- /lightning_ir/loss/listwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/listwise.py -------------------------------------------------------------------------------- /lightning_ir/loss/neural.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/neural.py -------------------------------------------------------------------------------- /lightning_ir/loss/pairwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/pairwise.py -------------------------------------------------------------------------------- /lightning_ir/loss/regularization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/loss/regularization.py -------------------------------------------------------------------------------- /lightning_ir/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/main.py -------------------------------------------------------------------------------- /lightning_ir/modeling_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lightning_ir/modeling_utils/batching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/modeling_utils/batching.py -------------------------------------------------------------------------------- /lightning_ir/modeling_utils/embedding_post_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/modeling_utils/embedding_post_processing.py -------------------------------------------------------------------------------- /lightning_ir/modeling_utils/lm_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/modeling_utils/lm_head.py -------------------------------------------------------------------------------- /lightning_ir/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/__init__.py -------------------------------------------------------------------------------- /lightning_ir/models/bi_encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/bi_encoders/__init__.py -------------------------------------------------------------------------------- /lightning_ir/models/bi_encoders/coil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/bi_encoders/coil.py -------------------------------------------------------------------------------- /lightning_ir/models/bi_encoders/col.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/bi_encoders/col.py -------------------------------------------------------------------------------- /lightning_ir/models/bi_encoders/dpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/bi_encoders/dpr.py -------------------------------------------------------------------------------- /lightning_ir/models/bi_encoders/mvr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/bi_encoders/mvr.py -------------------------------------------------------------------------------- /lightning_ir/models/bi_encoders/splade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/bi_encoders/splade.py -------------------------------------------------------------------------------- /lightning_ir/models/cross_encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/cross_encoders/__init__.py -------------------------------------------------------------------------------- /lightning_ir/models/cross_encoders/mono.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/cross_encoders/mono.py -------------------------------------------------------------------------------- /lightning_ir/models/cross_encoders/set_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/cross_encoders/set_encoder.py -------------------------------------------------------------------------------- /lightning_ir/models/register_external_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/register_external_models.py -------------------------------------------------------------------------------- /lightning_ir/models/register_internal_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/models/register_internal_models.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/__init__.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/base/__init__.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/base/indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/base/indexer.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/base/packed_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/base/packed_tensor.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/base/searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/base/searcher.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/faiss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/faiss/__init__.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/faiss/faiss_indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/faiss/faiss_indexer.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/faiss/faiss_searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/faiss/faiss_searcher.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/plaid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/plaid/__init__.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/plaid/plaid_indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/plaid/plaid_indexer.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/plaid/plaid_searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/plaid/plaid_searcher.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/pytorch/__init__.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/pytorch/dense_indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/pytorch/dense_indexer.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/pytorch/dense_searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/pytorch/dense_searcher.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/pytorch/sparse_indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/pytorch/sparse_indexer.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/pytorch/sparse_searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/pytorch/sparse_searcher.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/seismic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/seismic/__init__.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/seismic/seismic_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/seismic/seismic_format.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/seismic/seismic_indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/seismic/seismic_indexer.py -------------------------------------------------------------------------------- /lightning_ir/retrieve/seismic/seismic_searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/retrieve/seismic/seismic_searcher.py -------------------------------------------------------------------------------- /lightning_ir/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/schedulers/__init__.py -------------------------------------------------------------------------------- /lightning_ir/schedulers/lr_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/schedulers/lr_schedulers.py -------------------------------------------------------------------------------- /lightning_ir/schedulers/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/lightning_ir/schedulers/schedulers.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/data/corpus/docpairs.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/docpairs.tsv -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/docs.tsv -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv.pklz4/bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/docs.tsv.pklz4/bin -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv.pklz4/bin.meta: -------------------------------------------------------------------------------- 1 | doc_id text -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv.pklz4/bin.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/docs.tsv.pklz4/bin.pos -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv.pklz4/idx.doc_id.key: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/docs.tsv.pklz4/idx.doc_id.key -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv.pklz4/idx.doc_id.meta: -------------------------------------------------------------------------------- 1 | 11 20 -------------------------------------------------------------------------------- /tests/data/corpus/docs.tsv.pklz4/idx.doc_id.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/docs.tsv.pklz4/idx.doc_id.pos -------------------------------------------------------------------------------- /tests/data/corpus/qrels.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/qrels.tsv -------------------------------------------------------------------------------- /tests/data/corpus/queries.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/corpus/queries.tsv -------------------------------------------------------------------------------- /tests/data/runs/lightning-ir.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/runs/lightning-ir.tsv -------------------------------------------------------------------------------- /tests/data/runs/run.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/runs/run.jsonl -------------------------------------------------------------------------------- /tests/data/test_model/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/test_model/config.json -------------------------------------------------------------------------------- /tests/data/test_model/model.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/test_model/model.safetensors -------------------------------------------------------------------------------- /tests/data/test_model/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/test_model/special_tokens_map.json -------------------------------------------------------------------------------- /tests/data/test_model/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/test_model/tokenizer.json -------------------------------------------------------------------------------- /tests/data/test_model/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/test_model/tokenizer_config.json -------------------------------------------------------------------------------- /tests/data/test_model/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/data/test_model/vocab.txt -------------------------------------------------------------------------------- /tests/test_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_callbacks.py -------------------------------------------------------------------------------- /tests/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_config.py -------------------------------------------------------------------------------- /tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_data.py -------------------------------------------------------------------------------- /tests/test_external_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_external_datasets.py -------------------------------------------------------------------------------- /tests/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_loss.py -------------------------------------------------------------------------------- /tests/test_lsr_benchmark_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_lsr_benchmark_integration.py -------------------------------------------------------------------------------- /tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_model.py -------------------------------------------------------------------------------- /tests/test_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_models/test_coil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_coil.py -------------------------------------------------------------------------------- /tests/test_models/test_col.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_col.py -------------------------------------------------------------------------------- /tests/test_models/test_dpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_dpr.py -------------------------------------------------------------------------------- /tests/test_models/test_mono.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_mono.py -------------------------------------------------------------------------------- /tests/test_models/test_mvr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_mvr.py -------------------------------------------------------------------------------- /tests/test_models/test_set_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_set_encoder.py -------------------------------------------------------------------------------- /tests/test_models/test_splade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_models/test_splade.py -------------------------------------------------------------------------------- /tests/test_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_schedulers.py -------------------------------------------------------------------------------- /tests/test_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/webis-de/lightning-ir/HEAD/tests/test_tokenizer.py --------------------------------------------------------------------------------