add rankbatch to base import

webis-de · Dec 12, 2024 · 3c0b67c · 3c0b67c
1 parent 7f21aef
commit 3c0b67c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lightning_ir/__init__.py b/lightning_ir/__init__.py
@@ -35,6 +35,7 @@
     LightningIRDataModule,
     QueryDataset,
     QuerySample,
+    RankBatch,
     RankSample,
     RunDataset,
     SearchBatch,
@@ -184,6 +185,7 @@
     "QueryDataset",
     "QuerySample",
     "RankCallback",
+    "RankBatch",
     "RankNet",
     "RankSample",
     "ReRankCallback",