Merge pull request #55 from webis-de/mam10eks-patch-1

Update ranking.py
webis-de · Sep 2, 2024 · ea86402 · ea86402
2 parents 4212946 + 0f44377
commit ea86402
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/ir_axioms/axiom/argumentative.py b/ir_axioms/axiom/argumentative.py
@@ -141,6 +141,7 @@ def _sentence_length(
         document: RankedDocument,
 ) -> float:
     download_nltk_dependencies("punkt")
+    download_nltk_dependencies("punkt_tab")
     sentences = sent_tokenize(context.contents(document))
     if len(sentences) == 0:
         return nan

diff --git a/ir_axioms/modules/ranking.py b/ir_axioms/modules/ranking.py
@@ -47,13 +47,15 @@ def kwiksort(
         axiom,
         query,
         context,
-        vertices_left
+        vertices_left,
+        pivot_selection
     )
     vertices_right = kwiksort(
         axiom,
         query,
         context,
-        vertices_right
+        vertices_right,
+        pivot_selection
     )
 
     return [*vertices_left, pivot, *vertices_right]

diff --git a/tests/unit/util.py b/tests/unit/util.py
@@ -17,6 +17,7 @@ class MemoryIndexContext(IndexContext):
     # noinspection PyMethodMayBeStatic
     def __post_init__(self):
         download_nltk_dependencies("punkt")
+        download_nltk_dependencies("punkt_tab")
 
     def __hash__(self):
         return reduce(