Update retrieve.py

chatnoir-eu · Feb 27, 2023 · f89bfe8 · f89bfe8
1 parent d17359c
commit f89bfe8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/chatnoir_pyterrier/retrieve.py b/chatnoir_pyterrier/retrieve.py
@@ -137,6 +137,7 @@ def _transform_query(self, topic: DataFrame) -> DataFrame:
                 explain=explain,
                 staging=self.staging,
                 page_size=page_size,
+                retries=self.retries,
             ).results
         else:
             results = search_phrases(
@@ -147,6 +148,7 @@ def _transform_query(self, topic: DataFrame) -> DataFrame:
                 explain=explain,
                 staging=self.staging,
                 page_size=page_size,
+                retries=self.retries,
             ).results
 
         if self.filter_unknown: